| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | merge_cast_to_op.cc | 41 candidate_kernel_info->GetInputFormat(i) != AnfAlgo::GetInputFormat(node, i)) { in AlternativeKernelInfoForInput()
45 candidate_kernel_info->GetInputFormat(i) != AnfAlgo::GetInputFormat(node, i)) { in AlternativeKernelInfoForInput()
80 if (AnfAlgo::GetInputFormat(node, index) != kernel_info->GetInputFormat(index) || in CheckInputs()
|
| D | rectify_do_mask_kernel_info.cc | 68 auto do_mask_data_format = AnfAlgo::GetInputFormat(do_mask, 0); in RectifyKernelInfo()
112 if (AnfAlgo::GetInputFormat(do_mask, 0) != format) { in RectifyDropOutDoMaskKernelInfo()
133 auto do_mask_input_format = AnfAlgo::GetInputFormat(node, 0); in RectifyKernelInfoInPynativeProcess()
|
| D | change_axis_of_reduce_kernel.cc | 73 auto format = AnfAlgo::GetInputFormat(cnode, 0); in ConvertReduceAttrFraczAnd6HD()
109 auto convert_map = kReduceConvertMap.find(AnfAlgo::GetInputFormat(node, 0)); in Process()
|
| D | trans_op_format_refine.cc | 36 auto in_format = AnfAlgo::GetInputFormat(node, 0); in Process()
|
| D | insert_transdata_for_runop.cc | 40 auto input_format = AnfAlgo::GetInputFormat(cnode, index); in Run()
|
| D | insert_transpose_for_dynamic_gru_v2.cc | 53 auto input_format = AnfAlgo::GetInputFormat(transdata_node, 0); in Insert()
|
| D | check_consistency.cc | 35 string selected_input_format = AnfAlgo::GetInputFormat(node, input_index); in CheckFormatForConsistency()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | reorder_ops.cc | 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()
209 AnfAlgo::GetInputFormat(node, 0) != AnfAlgo::GetOutputFormat(node, 0)) { in ReorderTypeInsensitiveCastDown()
238 cast_io_info.inputs_format.push_back(AnfAlgo::GetInputFormat(type_insens_node, index)); in ReorderTypeInsensitiveCastDown()
277 AnfAlgo::GetInputFormat(node, 0) == AnfAlgo::GetOutputFormat(node, 0) && in ReorderCastUpTypeInsensitive()
|
| D | raise_reduction_precision.cc | 50 info_builder.SetInputsFormat({AnfAlgo::GetInputFormat(node, 0)}); in CreateReduceSum()
92 …auto cast1 = CreateCast(node->cast<CNodePtr>()->input(1), kFloat32, AnfAlgo::GetInputFormat(node, … in Process()
|
| D | substitute_dropout.cc | 76 …dropout_kernel_info_builder->SetInputsFormat({old_kernel_info->GetInputFormat(0), kOpFormat_DEFAUL… in PreProcess()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | transdata_split.cc | 46 auto input_format = AnfAlgo::GetInputFormat(node, 0); in IsFormatInvaild()
67 auto input_format = AnfAlgo::GetInputFormat(node, 0); in DoSplit()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.cc | 61 std::string input_format = AnfAlgo::GetInputFormat(trans_node, 0); in SetTransNodeAttr()
126 std::string dest_format = AnfAlgo::GetInputFormat(node, index); in GetTransInputNodePtr()
128 …MS_LOG(DEBUG) << node->DebugString() << "Insert transdata " << AnfAlgo::GetInputFormat(node, index) in GetTransInputNodePtr()
201 …std::string dst_format = is_insert_input ? AnfAlgo::GetInputFormat(node, insert_index) : default_f… in AddTransOpNodeToGraph()
218 …auto padding_shape = trans::PaddingShape(input_node_out_shape, AnfAlgo::GetInputFormat(node, inser… in AddTransOpNodeToGraph()
443 const std::string dev_fmt = AnfAlgo::GetInputFormat(cnode, input_index); in InsertCastForInput()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_select_ascend.cc | 153 …if (kernel_build_info.GetInputFormat(input_index) == AnfAlgo::GetPrevNodeOutputFormat(kernel_node,… in UpdateCurMatchCounts()
164 if (kernel_build_info.GetInputFormat(input_index) == pri_match_format) { in UpdateCurMatchCounts()
167 if (kernel_build_info.GetInputFormat(input_index) == kOpFormat_DEFAULT || in UpdateCurMatchCounts()
168 kernel_build_info.GetInputFormat(input_index) == kOpFormat_NCDHW) { in UpdateCurMatchCounts()
444 …if (RefreshCastAndParamWeightFormat(real_input_node, selected_kernel_info->GetInputFormat(input_in… in SetTensorDeviceInfo()
450 auto refresh_format = selected_kernel_info->GetInputFormat(input_index); in SetTensorDeviceInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | layer_norm_beta_gamma_backprop_fusion.cc | 60 kernel_info->GetInputFormat(i) != AnfAlgo::GetInputFormat(cnode, i)) { in CheckKernelBuildInfo()
|
| D | refresh_parameter_format.cc | 34 auto cnode_input_format = AnfAlgo::GetInputFormat(cnode, input_index); in DoRefresh()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_build_info.h | 53 std::string GetInputFormat(size_t input_index) const;
136 kernel_build_info_->inputs_format_.emplace_back(kernel_build_info->GetInputFormat(index)); in KernelBuildInfoBuilder()
|
| D | kernel_build_info.cc | 22 std::string KernelBuildInfo::GetInputFormat(size_t input_index) const { in GetInputFormat() function in mindspore::kernel::KernelBuildInfo
108 …output_buffer << "<" << ToShortString(GetInputDeviceType(index)) << "x" << GetInputFormat(index) <… in ToString()
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/include/api/ |
| D | context.h | 139 inline std::string GetInputFormat() const;
207 std::string Ascend310DeviceInfo::GetInputFormat() const { return CharToString(GetInputFormatChar())… in GetInputFormat() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | eliminate_redundant_op.cc | 77 return AnfAlgo::GetInputFormat(node1, 0) == AnfAlgo::GetOutputFormat(node2, 0) && in TransDataOpEliminateCondition()
78 AnfAlgo::GetOutputFormat(node1, 0) == AnfAlgo::GetInputFormat(node2, 0) && in TransDataOpEliminateCondition()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/cpu/ |
| D | insert_format_transform_op.cc | 103 …AnfAlgo::IsRealCNodeKernel(used_node) ? AnfAlgo::GetInputFormat(used_node, used_node_index) : kOpF… in ProcessForTupleItem()
162 …AnfAlgo::IsRealCNodeKernel(used_node) ? AnfAlgo::GetInputFormat(used_node, used_node_index) : kOpF… in InsertTransformOpForOutput()
|
| /third_party/mindspore/include/api/ |
| D | context.h | 319 inline std::string GetInputFormat() const;
415 std::string Ascend310DeviceInfo::GetInputFormat() const { return CharToString(GetInputFormatChar())… in GetInputFormat() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_attrs_process.cc | 186 auto left_format = AnfAlgo::GetInputFormat(anf_node, 0); in SetAkgAttrsForMatMul()
187 auto right_format = AnfAlgo::GetInputFormat(anf_node, 1); in SetAkgAttrsForMatMul()
|
| D | akg_kernel_json_generator.cc | 153 std::string AkgKernelJsonGenerator::GetInputFormat(const AnfNodePtr &anf_node, size_t real_index) c… in GetInputFormat() function in mindspore::kernel::AkgKernelJsonGenerator
154 …return dump_option_.is_before_select_kernel ? kOpFormat_DEFAULT : AnfAlgo::GetInputFormat(anf_node… in GetInputFormat()
202 input_desc_json[kJsonKeyFormat] = this->GetInputFormat(anf_node, real_input_index); in CreateInputDescJson()
286 auto input_format = this->GetInputFormat(anf_node, format_i); in GetAttrJson()
721 input_desc_json[kJsonKeyFormat] = this->GetInputFormat(tmp_input.first, tmp_input.second.first); in CreateInputsJson()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | kernel_info_setter.cc | 49 if (selected_kernel_info->GetInputFormat(i) != alternative_kernel_info->GetInputFormat(i)) { in CheckKernelInfo()
171 std::vector<std::string> output_format = {selected_kernel_info.GetInputFormat(input_index)}; in SetTensorDeviceInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | batch_norm_relu_fusion.cc | 54 if (AnfAlgo::GetInputFormat(batch_norm, 0) != kOpFormat_NHWC && format != "NHWC") { in Process()
|