Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/
Dms_device_shape_transfer.h420 std::vector<Axis5D> padding_axis; variable
437 std::vector<Axis> padding_axis; variable
487 std::vector<int> padding_axis; variable
Dms_device_shape_transfer.cc144 std::vector<Axis5D> padding_axis; local
183 std::vector<Axis> padding_axis; local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/
Dinsert_transpose_for_dynamic_gru_v2.cc58 auto padding_axis = AnfAlgo::GetOutputReshapeType(transdata_node, 0); in Insert() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dtransdata_split.cc97 auto padding_axis = AnfAlgo::GetOutputReshapeType(node, 0); in DoSplit() local