Home
last modified time | relevance | path

Searched defs:SetOpcontext (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/rpc/
Drpc_actor.cc21 void RpcActor::SetOpcontext(OpContext<DeviceTensor> *const op_context) { op_context_ = op_context; } in SetOpcontext() function in mindspore::runtime::RpcActor
Drecv_actor.cc42 void RecvActor::SetOpcontext(OpContext<DeviceTensor> *const op_context) { in SetOpcontext() function in mindspore::runtime::RecvActor
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Drpc_node_scheduler.cc186 void RpcNodeScheduler::SetOpcontext(const RpcActorSetPtr &rpc_actors, OpContext<DeviceTensor> *cons… in SetOpcontext() function in mindspore::runtime::RpcNodeScheduler