Searched defs:input_node_with_index (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/dynamic_shape/ |
| D | dynamic_shape_helper.cc | 111 …TensorPtr CreateTensorFromIndexedNode(const std::pair<AnfNodePtr, size_t> &input_node_with_index) { in CreateTensorFromIndexedNode()
156 tensor::TensorPtr CreateTensorMem(const std::pair<AnfNodePtr, size_t> &input_node_with_index, const… in CreateTensorMem()
186 … const std::pair<AnfNodePtr, size_t> &input_node_with_index, bool skip_nop_node, in GetDependValueTensor()
425 auto input_node_with_index = common::AnfAlgo::GetPrevNodeOutput(cnode, i, false); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/profiler/ |
| D | gpu_profiling_utils.cc | 137 auto input_node_with_index = common::AnfAlgo::GetPrevNodeOutput(*iter, i); in SetTraceBpEnd() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/rpc/ |
| D | recv_actor.cc | 286 auto input_node_with_index = common::AnfAlgo::GetPrevNodeOutput(kernel_, input_index, false); in AddArgSpecForInput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/ |
| D | deal_ref_output.cc | 235 auto input_node_with_index = common::AnfAlgo::GetPrevNodeOutput(cnode, i, true); in DealBroadCastAsRef() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 196 auto input_node_with_index = common::AnfAlgo::GetPrevNodeOutput(node, i, true); in GetCommunicationInputInfo() local
857 auto input_node_with_index = common::AnfAlgo::GetPrevNodeOutput(anf_node, index); in PreAssignCNodeMemory() local
878 auto input_node_with_index = common::AnfAlgo::GetPrevNodeOutput(node, i, true); in AssignCommunicationNodeInputMem() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/ |
| D | framework_utils.cc | 1055 auto input_node_with_index = common::AnfAlgo::GetPrevNodeOutput(node, i, false); in SetInputsByConstInputs() local
|