Home
last modified time | relevance | path

Searched defs:padding_axis (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/common/
Dtrans.h199 std::vector<Axis5D> padding_axis; variable
213 std::vector<Axis> padding_axis; variable
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dtransdata_split.cc72 auto padding_axis = AnfAlgo::GetOutputReshapeType(node, 0); in DoSplit() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dinsert_transpose_for_dynamic_gru_v2.cc55 auto padding_axis = AnfAlgo::GetOutputReshapeType(transdata_node, 0); in Insert() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_helper.cc202 std::string padding_axis = is_insert_input ? AnfAlgo::GetInputReshapeType(node, insert_index) in AddTransOpNodeToGraph() local
290 auto padding_axis = AnfAlgo::GetOutputReshapeType(input, 0); in NewTransOpNode() local