Searched refs:dev_fmt (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | insert_cast.cc | 62 const auto dev_fmt = AnfAlgo::GetOutputFormat(cnode, output_idx); in InsertCastForMultipleOutput() local
65 …replace_node = AddCastOpNodeToGraph(func_graph, getitem, dev_fmt, device_type, origin_type, origin… in InsertCastForMultipleOutput()
95 const std::string dev_fmt = AnfAlgo::GetOutputFormat(cnode, 0); in InsertCastForOutput() local
101 …replace_node = AddCastOpNodeToGraph(func_graph, cnode, dev_fmt, device_type, origin_type, origin_s… in InsertCastForOutput()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/cpu/ |
| D | insert_cast_cpu.cc | 83 const std::string dev_fmt = AnfAlgo::GetInputFormat(cnode, input_index); in InsertCast() local
88 …AddCastOpNodeToGraph(func_graph, cur_input, dev_fmt, origin_type, device_type, origin_shape, devic… in InsertCast()
102 const std::string dev_fmt = AnfAlgo::GetOutputFormat(cnode, i); in InsertCastForGraphOutput() local
115 …AddCastOpNodeToGraph(func_graph, cur_input, dev_fmt, device_type, infer_type, origin_shape, infer_… in InsertCastForGraphOutput()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | decrease_compute_precision.cc | 94 const std::string dev_fmt = AnfAlgo::GetInputFormat(cnode, input_index); in InsertCastForGraphKernel() local
109 …auto cast = AddCastCNode(func_graph, cur_input, dev_fmt, ori_dtype, new_dtype, ori_shape, new_dtyp… in InsertCastForGraphKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ |
| D | ascend_helper.cc | 443 const std::string dev_fmt = AnfAlgo::GetInputFormat(cnode, input_index); in InsertCastForInput() local
449 …AddCastOpNodeToGraph(func_graph, cur_input, dev_fmt, origin_type, device_type, origin_shape, infer… in InsertCastForInput()
|