| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/cpu/ |
| D | insert_format_transform_op.cc | 113 auto inputs_format = AnfAlgo::GetAllInputFormats(node); in InsertTransformOpForInput() local
114 for (size_t i = 0; i < inputs_format.size(); ++i) { in InsertTransformOpForInput()
115 if ((inputs_format[i] == kOpFormat_DEFAULT) || (inputs_format[i] == origin_format)) { in InsertTransformOpForInput()
119 if (inputs_format[i] == prev_input_format) { in InsertTransformOpForInput()
131 SetTransposeOpBuildInfo(kOpFormat_DEFAULT, inputs_format[i], transpose_op); in InsertTransformOpForInput()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | insert_format_transform_op.cc | 138 auto inputs_format = AnfAlgo::GetAllInputFormats(node); in Process() local
139 for (size_t i = 0; i < inputs_format.size(); i++) { in Process()
140 if ((inputs_format[i] != kOpFormat_DEFAULT) && (inputs_format[i] != origin_data_format)) { in Process()
143 auto input_transpose_perm = TransposeAxis(origin_data_format, inputs_format[i]); in Process()
145 SetTransposeOpBuildInfo(kOpFormat_DEFAULT, inputs_format[i], input_transpose_op); in Process()
|
| D | add_relu_v2_fusion.cc | 31 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
40 inputs_format.push_back(kOpFormat_DEFAULT); in GenerateKernelBuildInfo()
48 builder.SetInputsFormat(inputs_format); in GenerateKernelBuildInfo()
|
| D | matmul_biasadd_fusion.cc | 31 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
40 inputs_format.push_back(kOpFormat_DEFAULT); in GenerateKernelBuildInfo()
48 builder.SetInputsFormat(inputs_format); in GenerateKernelBuildInfo()
|
| D | add_relu_grad_v2_fusion.cc | 31 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
40 inputs_format.push_back(kOpFormat_DEFAULT); in GenerateKernelBuildInfo()
48 builder.SetInputsFormat(inputs_format); in GenerateKernelBuildInfo()
|
| D | relu_v2_pass.cc | 44 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
53 inputs_format.push_back(kOpFormat_DEFAULT); in GenerateKernelBuildInfo()
61 builder.SetInputsFormat(inputs_format); in GenerateKernelBuildInfo()
|
| D | adam_fusion.cc | 31 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
41 inputs_format.push_back(kOpFormat_DEFAULT); in GenerateKernelBuildInfo()
49 builder.SetInputsFormat(inputs_format); in GenerateKernelBuildInfo()
|
| D | adam_weight_decay_fusion.cc | 31 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
41 inputs_format.push_back(kOpFormat_DEFAULT); in GenerateKernelBuildInfo()
49 builder.SetInputsFormat(inputs_format); in GenerateKernelBuildInfo()
|
| D | print_reduce_fusion.cc | 30 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
38 inputs_format.push_back(kOpFormat_DEFAULT); in GenerateKernelBuildInfo()
47 builder.SetInputsFormat(inputs_format); in GenerateKernelBuildInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | reorder_ops.cc | 122 info_builder.SetInputsFormat(node_io_info.inputs_format); in SetNodeInfo()
182 new_inputs_info->inputs_format.resize(0); in SetTypeInsensitiveNodeInputsInfo()
188 new_inputs_info->inputs_format.push_back(AnfAlgo::GetInputFormat(node, data_idx)); in SetTypeInsensitiveNodeInputsInfo()
192 new_inputs_info->inputs_format.push_back(AnfAlgo::GetInputFormat(input_at_indexes[idx], 0)); in SetTypeInsensitiveNodeInputsInfo()
195 … new_inputs_info->inputs_format.push_back(AnfAlgo::GetOutputFormat(input_at_indexes[idx], 0)); in SetTypeInsensitiveNodeInputsInfo()
238 cast_io_info.inputs_format.push_back(AnfAlgo::GetInputFormat(type_insens_node, index)); in ReorderTypeInsensitiveCastDown()
239 cast_io_info.outputs_format = cast_io_info.inputs_format; in ReorderTypeInsensitiveCastDown()
313 cast_io_info.inputs_format.push_back(pattern_output_format); in ReorderCastUpTypeInsensitive()
314 cast_io_info.outputs_format = cast_io_info.inputs_format; in ReorderCastUpTypeInsensitive()
|
| D | graph_kernel_helper.h | 66 kernel::KernelBuildInfoPtr BuildSelectKernelBuildInfo(const std::vector<std::string> &inputs_format,
70 kernel::KernelBuildInfoPtr BuildSelectKernelBuildInfo(const std::vector<std::string> &inputs_format,
|
| D | reorder_ops.h | 28 std::vector<std::string> inputs_format; member
|
| D | graph_kernel_helper.cc | 287 kernel::KernelBuildInfoPtr BuildSelectKernelBuildInfo(const std::vector<std::string> &inputs_format, in BuildSelectKernelBuildInfo() argument
292 graph_info_builder.SetInputsFormat(inputs_format); in BuildSelectKernelBuildInfo()
303 kernel::KernelBuildInfoPtr BuildSelectKernelBuildInfo(const std::vector<std::string> &inputs_format, in BuildSelectKernelBuildInfo() argument
308 graph_info_builder.SetInputsFormat(inputs_format); in BuildSelectKernelBuildInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | host_kernel_metadata.cc | 41 std::vector<std::string> inputs_format{}; in HostMetadataInfo() local
45 inputs_format.emplace_back(kOpFormat_DEFAULT); in HostMetadataInfo()
56 builder.SetInputsFormat(inputs_format); in HostMetadataInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_metadata.cc | 54 std::vector<std::string> inputs_format{}; in AicpuMetadataInfoForSpecialNodes() local
60 inputs_format.emplace_back(kOpFormat_DEFAULT); in AicpuMetadataInfoForSpecialNodes()
72 builder.SetInputsFormat(inputs_format); in AicpuMetadataInfoForSpecialNodes()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | kernel_info_setter.cc | 248 … std::vector<std::string> *inputs_format, std::vector<std::string> *outputs_format, in UpdateKernelFormatInfo() argument
251 MS_EXCEPTION_IF_NULL(inputs_format); in UpdateKernelFormatInfo()
264 if (input_format_position >= inputs_format->size()) { in UpdateKernelFormatInfo()
266 … << inputs_format->size() << "] #kernel_node [" << kernel_node->fullname_with_scope() << "]"; in UpdateKernelFormatInfo()
268 (*inputs_format)[input_format_position] = cal_format; in UpdateKernelFormatInfo()
406 std::vector<std::string> inputs_format; in SetKernelInfo() local
410 inputs_format.emplace_back(kOpFormat_DEFAULT); in SetKernelInfo()
422 …UpdateKernelFormatInfo(kernel_node, inputs_type, &inputs_format, &outputs_format, &origin_data_for… in SetKernelInfo()
426 builder->SetInputsFormat(inputs_format); in SetKernelInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hccl_kernel_metadata.cc | 87 std::vector<std::string> inputs_format{}; in HcclMetadataInfo() local
91 (void)inputs_format.emplace_back(GetKernelFormat(kernel_node, input_index)); in HcclMetadataInfo()
106 builder.SetInputsFormat(inputs_format); in HcclMetadataInfo()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_launch_transdata.cc | 108 std::vector<std::string> inputs_format = {src_format_}; in ConstructKernelGraphAndSetAttr() local
110 builder->SetInputsFormat(inputs_format); in ConstructKernelGraphAndSetAttr()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/ |
| D | cpu_device_context.cc | 147 std::vector<std::string> inputs_format; in SetControlOpInfo() local
151 (void)inputs_format.emplace_back(kOpFormat_DEFAULT); in SetControlOpInfo()
163 builder->SetInputsFormat(inputs_format); in SetControlOpInfo()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/pass/ |
| D | common_subexpression_elimination_test.cc | 42 kernel::KernelBuildInfoPtr CreateKernelBuildInfo(const std::vector<std::string> &inputs_format, in CreateKernelBuildInfo() argument
48 builder1.SetInputsFormat(inputs_format); in CreateKernelBuildInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_build_info.cc | 156 …BuildInfo::KernelBuildInfoBuilder::SetInputsFormat(const std::vector<std::string> &inputs_format) { in SetInputsFormat() argument
158 kernel_build_info_->inputs_format_ = inputs_format; in SetInputsFormat()
|
| D | common_utils.cc | 306 std::vector<std::string> inputs_format; in SetInputKernelBuilderInfo() local
333 inputs_format.push_back(formats[builder_idex]); in SetInputKernelBuilderInfo()
340 inputs_format.push_back(formats[builder_idex]); in SetInputKernelBuilderInfo()
347 inputs_format.push_back(formats[builder_idex]); in SetInputKernelBuilderInfo()
353 builder->SetInputsFormat(inputs_format); in SetInputKernelBuilderInfo()
|
| D | kernel_build_info.h | 153 void SetInputsFormat(const std::vector<std::string> &inputs_format);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_kernel_select.cc | 108 std::vector<std::string> inputs_format; in GetCommonPatternKernelInfo() local
114 … &inputs_format, &inputs_device_type, &inputs_reshape_type, &inputs_value_depend)) { in GetCommonPatternKernelInfo()
118 builder.SetInputsFormat(inputs_format); in GetCommonPatternKernelInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | ub_pattern_fusion.cc | 113 std::vector<std::string> inputs_format; in CreateFusionOpKernelInfo() local
117 (void)inputs_format.emplace_back(AnfAlgo::GetOutputFormat(real_input.first, real_input.second)); in CreateFusionOpKernelInfo()
136 builder.SetInputsFormat(inputs_format); in CreateFusionOpKernelInfo()
|