Home
last modified time | relevance | path

Searched defs:input_with_index (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dinline_control_flow_scheduler.cc407 void InlineControlFlowScheduler::FixRefCountForRefNode(const KernelWithIndex &input_with_index, siz… in FixRefCountForRefNode()
459 void InlineControlFlowScheduler::FixRefCountForInputNode(const KernelWithIndex &input_with_index, s… in FixRefCountForInputNode()
537 const auto &input_with_index = in FixRefCountByConditionGatherActor() local
Dcontrol_node_parser.cc1954 const auto &input_with_index = input_with_indexs[i]; in CreateDeviceTensors() local
2447 …[&depend_nodes](const auto &input_with_index) { (void)depend_nodes.emplace(input_with_index.first)… in IsCallNodeNeedStack()
2485 for (auto input_with_index : input_with_indexs) { in ParseNeedStackControlNode() local
2795 for (const auto &input_with_index : input_with_indexes) { in IsInputInSameLevel() local
Dcontrol_node_scheduler.cc749 auto input_with_index = common::AnfAlgo::VisitKernelWithReturnType(input_node, 0, false); in FetchInternalParameterInput() local
1935 auto input_with_index = common::AnfAlgo::VisitKernelWithReturnType(input_node, 0, false); in LinkDataArrowByKernelGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Dcommon_subexpression_elimination.cc68 KernelWithIndex input_with_index{getitem_cnode->input(kRealInputNodeIndexInTupleGetItem), in EliminateDuplicatedTupleGetItem() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/
Dkernel_select_ascend.cc173 auto input_with_index = common::AnfAlgo::VisitKernelWithReturnType(input_kernel_node, 0); in RefreshInputParameter() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/
Dkernel_info_setter.cc269 auto input_with_index = common::AnfAlgo::VisitKernel(input_kernel_node, 0); in SetTensorDeviceInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dge_graph_executor.cc590 auto input_with_index = std::make_pair(input_node, 0); in AllocInputHostMemory() local
842 auto input_with_index = ref_map[output_with_index]; in CreateOutputDeviceAddress() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/data_dump/
Ddump_json_parser.cc1143 auto input_with_index = common::AnfAlgo::GetPrevNodeOutput(kernel, i); in UpdateNeedDumpKernels() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Danf_runtime_algorithm.cc1521 auto input_with_index = common::AnfAlgo::GetPrevNodeOutput(node, i); in SkipDataSync() local
1548 auto input_with_index = common::AnfAlgo::GetPrevNodeOutput(node, i); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Danfalgo.cc1920 auto input_with_index = GetPrevNodeOutput(node, i); in IsNodeInputContainMonad() local