Home
last modified time | relevance | path

Searched refs:GetOutputFormat (Results 1 – 25 of 65) sorted by relevance

123

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dmerge_cast_to_op.cc53 candidate_kernel_info->GetOutputFormat(i) != AnfAlgo::GetOutputFormat(node, i)) { in AlternativeKernelInfoForInput()
98 if (AnfAlgo::GetOutputFormat(node, index) != kernel_info->GetOutputFormat(index) || in CheckOtherOutputs()
116 …AnfAlgo::GetOutputFormat(node, 0) == kOpFormat_NCHW && kernel_info->GetOutputFormat(index) == kOpF… in CheckIndexOutput()
119 return AnfAlgo::GetOutputFormat(node, 0) == kernel_info->GetOutputFormat(index); in CheckIndexOutput()
Ddeal_ref_and_split_unsupported_transdata.cc114 auto origin_format = AnfAlgo::GetOutputFormat(origin_pair.first, origin_pair.second); in AddAdditionalToRefOutput()
116 auto cur_format = AnfAlgo::GetOutputFormat(cnode, output_index); in AddAdditionalToRefOutput()
280 kHWSpecialFormatSet.find(kernel_info->GetOutputFormat(0)) == kHWSpecialFormatSet.end()) { in SplitTransdataIfNotSupported()
303 …RefreshKernelBuildInfo(kOpFormat_DEFAULT, AnfAlgo::GetOutputFormat(next_trans_node, 0), next_trans… in SplitTransdataIfNotSupported()
Dinsert_cast.cc62 const auto dev_fmt = AnfAlgo::GetOutputFormat(cnode, output_idx); in InsertCastForMultipleOutput()
95 const std::string dev_fmt = AnfAlgo::GetOutputFormat(cnode, 0); in InsertCastForOutput()
Dtrans_op_format_refine.cc37 auto out_format = AnfAlgo::GetOutputFormat(node, 0); in Process()
Dcheck_consistency.cc34 string pre_output_format = AnfAlgo::GetOutputFormat(prev_node.first, prev_node.second); in CheckFormatForConsistency()
Dinsert_transpose_for_dynamic_gru_v2.cc54 auto output_format = AnfAlgo::GetOutputFormat(transdata_node, 0); in Insert()
/third_party/googletest/googletest/test/
Dgoogletest-options-test.cc65 EXPECT_STREQ("", UnitTestOptions::GetOutputFormat().c_str()); in TEST()
68 TEST(XmlOutputTest, GetOutputFormat) { in TEST() argument
70 EXPECT_STREQ("xml", UnitTestOptions::GetOutputFormat().c_str()); in TEST()
/third_party/mindspore/tests/ut/cpp/session/
Dkernel_graph_test.cc55 EXPECT_EQ(AnfAlgo::GetOutputFormat(new_value, 0), kOpFormat_DEFAULT); in TEST_F()
66 EXPECT_EQ(AnfAlgo::GetOutputFormat(new_paramter, 0), kOpFormat_DEFAULT); in TEST_F()
79 EXPECT_EQ(AnfAlgo::GetOutputFormat(new_non_weight_parameter, 0), kOpFormat_DEFAULT); in TEST_F()
93 EXPECT_EQ(AnfAlgo::GetOutputFormat(new_weight_parameter_node, 0), kOpFormat_DEFAULT); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/reporter/
Dgraph_desc_reporter.cc42 element.data_format_ = AnfAlgo::GetOutputFormat(input_node, input_index); in ReportData()
52 element.data_format_ = AnfAlgo::GetOutputFormat(node, i); in ReportData()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dreorder_ops.cc195 … new_inputs_info->inputs_format.push_back(AnfAlgo::GetOutputFormat(input_at_indexes[idx], 0)); in SetTypeInsensitiveNodeInputsInfo()
209 AnfAlgo::GetInputFormat(node, 0) != AnfAlgo::GetOutputFormat(node, 0)) { in ReorderTypeInsensitiveCastDown()
215 auto pattern_output_format = AnfAlgo::GetOutputFormat(node, 0); in ReorderTypeInsensitiveCastDown()
277 AnfAlgo::GetInputFormat(node, 0) == AnfAlgo::GetOutputFormat(node, 0) && in ReorderCastUpTypeInsensitive()
289 auto pattern_output_format = AnfAlgo::GetOutputFormat(node, 0); in ReorderCastUpTypeInsensitive()
Ddecrease_transfer_precision.cc162 std::vector<std::string> cnode_input_format = {AnfAlgo::GetOutputFormat(old_output, 0)}; in Process_Father()
164 std::vector<std::string> cnode_output_format = {AnfAlgo::GetOutputFormat(old_output, 0)}; in Process_Father()
275 std::vector<std::string> cnode_input_format = {AnfAlgo::GetOutputFormat(tensor_input, 0)}; in Process_Son()
277 std::vector<std::string> cnode_output_format = {AnfAlgo::GetOutputFormat(tensor_input, 0)}; in Process_Son()
Draise_reduction_precision.cc52 info_builder.SetOutputsFormat({AnfAlgo::GetOutputFormat(node, 0)}); in CreateReduceSum()
94 auto cast2 = CreateCast(new_reduce, kFloat16, AnfAlgo::GetOutputFormat(node, 0)); in Process()
Dtsa_atomic_add_to_first_tensor.cc115 parameter_format = AnfAlgo::GetOutputFormat(kernel_with_index.first, kernel_with_index.second); in ProcessTsaFirstNode()
170 AnfAlgo::GetOutputFormat(kernel_with_index.first, kernel_with_index.second); in CorrectKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_helper.cc62 std::string output_format = AnfAlgo::GetOutputFormat(trans_node, 0); in SetTransNodeAttr()
143 std::string output_format = AnfAlgo::GetOutputFormat(node, 0); in InsertTransOpForSingleOutput()
170 std::string output_format = AnfAlgo::GetOutputFormat(node, output_idx); in InsertTransOpForMultipleOutput()
200 …std::string input_format = is_insert_input ? default_format : AnfAlgo::GetOutputFormat(node, inser… in AddTransOpNodeToGraph()
293 …ShapeVector pad_shape = trans::PaddingShape(out_shape, AnfAlgo::GetOutputFormat(input, 0), padding… in NewTransOpNode()
295 …ShapeVector pad_shape_min = trans::PaddingShape(out_shape_min, AnfAlgo::GetOutputFormat(input, 0),… in NewTransOpNode()
296 …ShapeVector pad_shape_max = trans::PaddingShape(out_shape_max, AnfAlgo::GetOutputFormat(input, 0),… in NewTransOpNode()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dtransdata_split.cc47 auto output_format = AnfAlgo::GetOutputFormat(node, 0); in IsFormatInvaild()
68 auto output_format = AnfAlgo::GetOutputFormat(node, 0); in DoSplit()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dparameter_and_transop_fusion.cc112 auto format = AnfAlgo::GetOutputFormat(cur_transop, 0); in Run()
114 auto param_format = AnfAlgo::GetOutputFormat(final_node, 0); in Run()
Dlayer_norm_beta_gamma_backprop_fusion.cc66 kernel_info->GetOutputFormat(i) != AnfAlgo::GetOutputFormat(cnode, i)) { in CheckKernelBuildInfo()
Drefresh_parameter_format.cc35 auto kernel_node_format = AnfAlgo::GetOutputFormat(input_kernel_node, 0); in DoRefresh()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_build_info.h55 std::string GetOutputFormat(size_t output_index) const;
142 kernel_build_info_->outputs_format_.emplace_back(kernel_build_info->GetOutputFormat(index)); in KernelBuildInfoBuilder()
Dkernel_build_info.cc30 std::string KernelBuildInfo::GetOutputFormat(size_t output_index) const { in GetOutputFormat() function in mindspore::kernel::KernelBuildInfo
115 …output_buffer << "<" << ToShortString(GetOutputDeviceType(index)) << "x" << GetOutputFormat(index)… in ToString()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Deliminate_redundant_op.cc77 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/runtime/device/ascend/
Dkernel_select_graph_kernel.cc210 ++all_input_formats[AnfAlgo::GetOutputFormat(para, 0)]; in GetDefaultFormat()
285 graph_input_format->push_back(AnfAlgo::GetOutputFormat(para, 0)); in UpdateInputsKernelInfo()
322 auto out_format = AnfAlgo::GetOutputFormat(cnode, 0); in UpdateEquivFormat()
456 graph_output_format.push_back(AnfAlgo::GetOutputFormat(output.first, output.second)); in SetGraphKernelInfo()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_runtime.cc295 …auto device_address = CreateDeviceAddress(nullptr, tensor_size, AnfAlgo::GetOutputFormat(input_nod… in ResetNodeAddress()
302 auto output_format = AnfAlgo::GetOutputFormat(kernel, i); in ResetNodeAddress()
424 …CreateDeviceAddress(nullptr, tensor_size, AnfAlgo::GetOutputFormat(item, index), output_type_id, {… in RunOpAssignInputMemory()
471 std::string output_format = AnfAlgo::GetOutputFormat(kernel, i); in RunOpAssignOutputMemory()
612 …device_address = CreateDeviceAddress(address.addr, address.size, AnfAlgo::GetOutputFormat(item, in… in AssignStaticMemoryInput()
620 …CreateDeviceAddress(nullptr, tensor_size, AnfAlgo::GetOutputFormat(item, index), output_type_id, {… in AssignStaticMemoryInput()
789 std::string output_format = AnfAlgo::GetOutputFormat(node, j); in AssignCommunicationNodeOutputMem()
829 std::string output_format = AnfAlgo::GetOutputFormat(anf_node, index); in PreAssignCNodeMemory()
922 std::string output_format = AnfAlgo::GetOutputFormat(node, i); in AssignNodeOutputMem()
977 auto output_format = AnfAlgo::GetOutputFormat(value_node, output_idx); in AssignValueNodeTensor()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.cc63 if (selected_kernel_info->GetOutputFormat(i) != alternative_kernel_info->GetOutputFormat(i)) { in CheckKernelInfo()
326 graph_output_format.push_back(AnfAlgo::GetOutputFormat(output.first, output.second)); in SetGraphKernelInfo()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.cc100 … AnfAlgo::GetOutputFormat(item, index), output_type_id); in CreateParameterDeviceAddress()
137 std::string output_format = AnfAlgo::GetOutputFormat(value_node, output_idx); in CreateDeviceAddressForTensorValue()
187 std::string output_format = AnfAlgo::GetOutputFormat(kernel, i); in CreateKernelOutputDeviceAddress()

123