Home
last modified time | relevance | path

Searched defs:op_context (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Dmemory_manager_actor.cc26 … const DeviceContext *device_context, OpContext<DeviceTensor> *const op_context, in AllocateMemory()
52 … OpContext<DeviceTensor> *const op_context, const AID &from_aid) { in AllocateContinuousMemory()
84 … OpContext<DeviceTensor> *const op_context, const AID &from_aid) { in AllocateBatchMemory()
136 OpContext<DeviceTensor> *const op_context) { in FreeBatchMemory()
165 void MemoryManagerActor::Wait(OpContext<DeviceTensor> *const op_context, const AID &from_aid) { in Wait()
172 OpContext<DeviceTensor> *const op_context) { in SetOpContextMemoryAllocFail()
Dactor_common.h62 #define SET_OPCONTEXT_FAIL_RET_WITH_ERROR(op_context, message) \ argument
69 #define SET_OPCONTEXT_SUCCESS_RET(op_context) \ argument
75 #define SET_OPCONTEXT_FAIL_RET_WITH_ERROR_BY_STRATEGY(strategy, op_context, message) \ argument
85 #define SET_OPCONTEXT_MEMORY_ALLOC_FAIL_BY_STRATEGY(strategy, op_context, device_context, kernel_na… argument
Ddebug_actor.cc35 … const DeviceContext *device_context, OpContext<DeviceTensor> *const op_context, in Debug()
77 OpContext<DeviceTensor> *const op_context, const AID *from_aid) { in DebugOnStepBegin()
102 void DebugActor::DebugOnStepEnd(OpContext<DeviceTensor> *const op_context, const AID *from_aid) { in DebugOnStepEnd()
Drecorder_actor.cc29 … const DeviceContext *device_context, OpContext<DeviceTensor> *const op_context) { in RecordInfo()
49 void RecorderActor::RecordOnStepEnd(OpContext<DeviceTensor> *const op_context) { in RecordOnStepEnd()
/third_party/mindspore/mindspore/_extends/parallel_compile/tbe_compiler/
Dcompiler.py124 import tbe.common.context.op_context as op_context namespace
Dtbe_process.py24 import tbe.common.context.op_context as op_context namespace
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_scheduler.cc330 OpContext<DeviceTensor> op_context; in Run() local