Home
last modified time | relevance | path

Searched defs:broadcast_shape_back (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/utils/
Dbroadcast_iterator.cc98 std::vector<int64_t> broadcast_shape_back; in GetBroadcastShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/
Dmasked_select_gpu_kernel.cc39 std::vector<int64_t> broadcast_shape_back; in GetBroadcastShape() local
Dmasked_select_grad_gpu_kernel.cc41 std::vector<int64_t> broadcast_shape_back; in GetBroadcastShape() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dget_tuple_index_info.cc38 ShapeVector broadcast_shape_back; in BroadCastShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dcpu_kernel.cc499 ShapeVector broadcast_shape_back; in GetBroadcastShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/
Dtensor_index_py.cc133 ShapeVector broadcast_shape_back; in BroadCastShape() local