Home
last modified time | relevance | path

Searched defs:input_kernel_node (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Drefresh_parameter_format.cc29 … auto input_kernel_node = AnfAlgo::VisitKernel(AnfAlgo::GetInputNode(cnode, input_index), 0).first; in DoRefresh() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_select_graph_kernel.cc64 auto input_kernel_node = AnfAlgo::GetInputNode(kernel_node, input_index); in ResetKernelBuildInfo() local
202 auto input_kernel_node = AnfAlgo::VisitKernel(kernel_node->input(i + 1), 0).first; in GetDefaultFormat() local
230 auto input_kernel_node = AnfAlgo::VisitKernel(kernel_node->input(i + 1), 0).first; in GetDefaultFormat() local
254 auto input_kernel_node = AnfAlgo::VisitKernel(kernel_node->input(i + 1), 0).first; in UpdateInputsKernelInfo() local
Dkernel_select_ascend.cc438 auto input_kernel_node = AnfAlgo::GetInputNode(kernel_node, input_index); in SetTensorDeviceInfo() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.cc148 auto input_kernel_node = kernel_node->input(input_index + 1); in SetTensorDeviceInfo() local