Home
last modified time | relevance | path

Searched refs:RefreshKernelBuildInfo (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dtransdata_split.cc78 RefreshKernelBuildInfo(input_format, kOpFormat_HWCN, new_transdata_node, padding_axis); in DoSplit()
82 RefreshKernelBuildInfo(kOpFormat_HWCN, output_format, new_transpose_node); in DoSplit()
91 RefreshKernelBuildInfo(input_format, kOpFormat_HWCN, new_transpose_node); in DoSplit()
96 RefreshKernelBuildInfo(kOpFormat_HWCN, output_format, new_transdata_node, padding_axis); in DoSplit()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dinsert_transpose_for_dynamic_gru_v2.cc63 RefreshKernelBuildInfo(input_format, kOpFormat_HWCN, new_transpose_node); in Insert()
68 RefreshKernelBuildInfo(kOpFormat_HWCN, output_format, new_transdata_node, padding_axis); in Insert()
Ddeal_ref_and_split_unsupported_transdata.cc124 RefreshKernelBuildInfo(cur_format, origin_format, final_node, {}, cur_type); in AddAdditionalToRefOutput()
302 RefreshKernelBuildInfo(AnfAlgo::GetInputFormat(cnode, 0), kOpFormat_DEFAULT, cnode); in SplitTransdataIfNotSupported()
303RefreshKernelBuildInfo(kOpFormat_DEFAULT, AnfAlgo::GetOutputFormat(next_trans_node, 0), next_trans… in SplitTransdataIfNotSupported()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_helper.h102 void RefreshKernelBuildInfo(const std::string &input_format, const std::string &output_format,
Dascend_helper.cc236 RefreshKernelBuildInfo(input_format, dst_format, trans_data, padding_axis); in AddTransOpNodeToGraph()
243 void RefreshKernelBuildInfo(const std::string &input_format, const std::string &output_format, in RefreshKernelBuildInfo() function