Overview
Operator execution context.
Provides runtime support such as input/output management, memory allocation, and stream control for operators running on devices.
Header File to Be Included
1 | #include <exe_graph/runtime/eager_op_execution_context.h> |
Public Member Functions
rtStream GetStream() const 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 Tensor *MallocOutputTensor(size_t index, const StorageShape &shape, const StorageFormat &format, ge::DataType dtype, size_t tensor_size) Tensor *MakeOutputRefInput(size_t output_index, size_t input_index) const void *MallocWorkSpace(size_t size) const Tensor *GetOutputTensor(size_t index) const
Parent topic: EagerOpExecutionContext