Overview
Context class in the operator build phase, which is inherited from ExtendedKernelContext. It provides capabilities such as obtaining input tensors (of the REQUIRED, OPTIONAL, or DYNAMIC type) required during build, querying build configuration options, and obtaining platform information. It is used to transfer the build context of the custom operator Compile API.
Header Files to Be Included
1 | #include <exe_graph/runtime/op_compile_context.h
|
Public Member Functions
const Tensor *GetInputTensor(size_t index) const const Tensor *GetRequiredInputTensor(size_t ir_index) const const Tensor *GetOptionalInputTensor(size_t ir_index) const const Tensor *GetDynamicInputTensor(size_t ir_index, size_t relative_index) const ge::graphStatus GetOption(const ge::AscendString &option_key, ge::AscendString &option) const ge::graphStatus GetPlatformInfos(fe::PlatFormInfos &platform_info, fe::OptionalInfos& optional_infos) const
Parent topic: OpCompileContext