Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dtranspose_impl.cu121 int input_tensor_index[3]{0, 0, 0}; in Transpose3DTensorSimple() local
242 int input_tensor_index[3]{0, 0, 0}; in Transpose3DTensorSimpleVector() local
257 int input_tensor_index[3]{0, 0, 0}; in Transpose3DTensorSimpleVector() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dsub_graph_split.cc598 for (uint32_t input_tensor_index : node->input_indices_) { in InitMiddleSubgraph() local
869 for (uint32_t input_tensor_index : node->input_indices_) { in SubGraphSplitByOffLineParallel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Ddata_prepare_actor.cc671 auto input_tensor_index = FetchInputTensorIndex(front_node); in FetchInputTensorByArg() local