Searched refs:input_with_index (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | kernel_info_setter.cc | 150 auto input_with_index = AnfAlgo::VisitKernel(input_kernel_node, 0); in SetTensorDeviceInfo() local
151 MS_EXCEPTION_IF_NULL(input_with_index.first); in SetTensorDeviceInfo()
152 auto real_input_node = input_with_index.first; in SetTensorDeviceInfo()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_select_ascend.cc | 440 auto input_with_index = AnfAlgo::VisitKernelWithReturnType(input_kernel_node, 0); in SetTensorDeviceInfo() local
441 MS_EXCEPTION_IF_NULL(input_with_index.first); in SetTensorDeviceInfo()
442 auto real_input_node = input_with_index.first; in SetTensorDeviceInfo()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/dump/ |
| D | data_dumper.cc | 82 auto input_with_index = AnfAlgo::GetPrevNodeOutput(kernel, i); in GetNeedDumpKernelList() local
83 auto input = input_with_index.first; in GetNeedDumpKernelList()
|
| /third_party/mindspore/mindspore/ccsrc/debug/data_dump/ |
| D | dump_json_parser.cc | 597 auto input_with_index = AnfAlgo::GetPrevNodeOutput(kernel, i); in UpdateNeedDumpKernels() local
598 auto input = input_with_index.first; in UpdateNeedDumpKernels()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_scheduler.cc | 1711 const auto &input_with_index = AnfAlgo::VisitKernelWithReturnType(inputs[i], 0); in LinkArrowByControlNode() local
1712 LinkDataArrowByControlNode(graph_compiler_info, input_with_index, from_func_graph, actor, in LinkArrowByControlNode()
1744 const auto &input_with_index = gather_actor->data_nodes_[i]; in LinkArrowByControlNode() local
1746 …LinkDataArrowByControlNode(graph_compiler_info, input_with_index, from_func_graph, gather_actor, i… in LinkArrowByControlNode()
1833 const KernelWithIndex &input_with_index, in LinkDataArrowByControlNode() argument
1838 const auto &input_node = input_with_index.first; in LinkDataArrowByControlNode()
1842 …LinkDataArrowByCallInput(input_with_index, graph_compiler_info.control_node_parser_, from_func_gra… in LinkDataArrowByControlNode()
1878 …auto op_arrow = std::make_shared<DataArrow>(input_with_index.second, to_actor->GetAID(), to_index); in LinkDataArrowByControlNode()
1881 …auto device_tensor = AnfAlgo::GetMutableOutputAddr(from_actor->kernel_, input_with_index.second, f… in LinkDataArrowByControlNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 2074 auto input_with_index = AnfAlgo::GetPrevNodeOutput(anf_node_ptr, i); in IsNodeInputDynamicShape() local
2075 auto input = input_with_index.first; in IsNodeInputDynamicShape()
2076 auto index = input_with_index.second; in IsNodeInputDynamicShape()
2229 auto input_with_index = AnfAlgo::GetPrevNodeOutput(node, i); in InferShape() local
2230 auto real_input = input_with_index.first; in InferShape()
2337 auto input_with_index = GetPrevNodeOutput(node, i); in IsNodeInputContainMonad() local
2338 if (HasAbstractMonad(input_with_index.first)) { in IsNodeInputContainMonad()
|