Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dget_tuple_index_info.cc118 static std::vector<ShapeVector> GetSliceShape(const std::vector<int64_t> &tuple_index_types, in GetSliceShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Dsplit_layernorm_comm_fp.cc60 ShapeVector GetSliceShape(const ShapeVector &origin_shape, size_t slice_axis = 0, int64_t slice_siz… in GetSliceShape() function
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/tensor_layout/
Dtensor_layout_test.cc216 TEST_F(TestTensorLayout, GetSliceShape) { in TEST_F() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.cc1285 Shape GetSliceShape(const Shape &tensor_shape, const Dimensions &strategy) { in GetSliceShape() function