Searched defs:tensors_mask (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | cpu_session.cc | 225 const std::vector<int64_t> &tensors_mask) { in BuildOpImpl()
275 const std::vector<int64_t> &tensors_mask) { in RunOpImplOrigin()
281 const std::vector<int64_t> &tensors_mask) { in RunOpImpl()
|
| D | session_basic.h | 232 const std::vector<int64_t> &tensors_mask) { in BuildOpImpl()
237 const std::vector<int64_t> &tensors_mask) {} in RunOpImpl()
240 const std::vector<int64_t> &tensors_mask) {} in RunOpImplOrigin()
|
| D | gpu_session.cc | 640 const std::vector<int64_t> &tensors_mask) { in BuildOpImpl()
665 const std::vector<int64_t> &tensors_mask) { in RunOpImplOrigin()
671 const std::vector<int64_t> &tensors_mask) { in RunOpImpl()
|
| D | pynative_task_manager.h | 33 …:string graph_info, bool is_dynamic_shape, KernelGraphPtr graph, std::vector<int64_t> tensors_mask, in RunOpContext()
|
| D | ascend_session.cc | 794 const std::vector<int64_t> &tensors_mask) { in BuildOpImpl()
916 … const std::vector<int64_t> &tensors_mask, bool cache_miss) { in CreateKernelGraph()
944 const std::vector<int64_t> &tensors_mask) { in RunOpImpl()
987 const std::vector<int64_t> &tensors_mask) { in RunOpImplOrigin()
1025 const std::vector<int64_t> &tensors_mask) { in PreBuildOp()
|
| D | executor.cc | 426 const std::vector<int64_t> &tensors_mask) { in RunOp()
|
| D | session_basic.cc | 2127 … const std::vector<int64_t> &tensors_mask, in ConstructSingleOpGraph()
2231 const std::vector<int64_t> &tensors_mask) { in RunOp()
2317 void SessionBasic::EraseValueNodeTensor(const std::vector<int64_t> &tensors_mask, in EraseValueNodeTensor()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | backend.cc | 471 const std::vector<int64_t> *tensors_mask, in CompileGraph()
984 std::vector<std::vector<int64_t> *> tensors_mask; in ConstructGraphCompilerInfo() local
992 const ActorInfo &actor_info, const std::vector<int64_t> *tensors_mask, in ConstructGraphCompilerInfo()
1037 const std::vector<int64_t> *tensors_mask, in RunGraph()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_compiler.cc | 385 const std::vector<int64_t> *tensors_mask, in CompileGraph()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute.cc | 309 const std::vector<int64_t> &tensors_mask, std::string *graph_info_key) { in GetSingleOpGraphInfo()
497 void ConstructInputTensor(const OpExecInfoPtr &op_run_info, std::vector<int64_t> *tensors_mask, in ConstructInputTensor()
1944 std::vector<int64_t> tensors_mask; in RunOpInMs() local
|