Searched refs:input_kernel_node (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | refresh_parameter_format.cc | 29 … auto input_kernel_node = AnfAlgo::VisitKernel(AnfAlgo::GetInputNode(cnode, input_index), 0).first; in DoRefresh() local
30 MS_EXCEPTION_IF_NULL(input_kernel_node); in DoRefresh()
31 if (input_kernel_node->isa<Parameter>()) { in DoRefresh()
35 auto kernel_node_format = AnfAlgo::GetOutputFormat(input_kernel_node, 0); in DoRefresh()
36 auto dtype = AnfAlgo::GetOutputDeviceDataType(input_kernel_node, 0); in DoRefresh()
40 AnfAlgo::SetSelectKernelBuildInfo(builder->Build(), input_kernel_node.get()); in DoRefresh()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_select_graph_kernel.cc | 64 auto input_kernel_node = AnfAlgo::GetInputNode(kernel_node, input_index); in ResetKernelBuildInfo() local
65 MS_EXCEPTION_IF_NULL(input_kernel_node); in ResetKernelBuildInfo()
66 auto kernel_with_index = AnfAlgo::VisitKernel(input_kernel_node, 0); in ResetKernelBuildInfo()
74 AnfAlgo::SetSelectKernelBuildInfo(builder.Build(), input_kernel_node.get()); in ResetKernelBuildInfo()
202 auto input_kernel_node = AnfAlgo::VisitKernel(kernel_node->input(i + 1), 0).first; in GetDefaultFormat() local
203 MS_EXCEPTION_IF_NULL(input_kernel_node); in GetDefaultFormat()
204 if (!input_kernel_node->isa<Parameter>()) { in GetDefaultFormat()
208 auto para = input_kernel_node->cast<ParameterPtr>(); in GetDefaultFormat()
230 auto input_kernel_node = AnfAlgo::VisitKernel(kernel_node->input(i + 1), 0).first; in GetDefaultFormat() local
231 MS_EXCEPTION_IF_NULL(input_kernel_node); in GetDefaultFormat()
[all …]
|
| D | kernel_select_ascend.cc | 438 auto input_kernel_node = AnfAlgo::GetInputNode(kernel_node, input_index); in SetTensorDeviceInfo() local
439 MS_EXCEPTION_IF_NULL(input_kernel_node); in SetTensorDeviceInfo()
440 auto input_with_index = AnfAlgo::VisitKernelWithReturnType(input_kernel_node, 0); in SetTensorDeviceInfo()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | kernel_info_setter.cc | 148 auto input_kernel_node = kernel_node->input(input_index + 1); in SetTensorDeviceInfo() local
149 MS_EXCEPTION_IF_NULL(input_kernel_node); in SetTensorDeviceInfo()
150 auto input_with_index = AnfAlgo::VisitKernel(input_kernel_node, 0); in SetTensorDeviceInfo()
|