Home
last modified time | relevance | path

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

/third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/
Dtensor_layout_test.cc216 TEST_F(TestTensorLayout, GetSliceShape) { in TEST_F() argument
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.h299 Shape GetSliceShape(const Shape &tensor_shape, const Dimensions &strategy);
Dconv2d_info.cc351 input_slice_shape_ = GetSliceShape(inputs_shape_[0], stra[0]); in InferDevMatrixShape()
988 input_slice_shape_ = GetSliceShape(inputs_shape_[0], stra[0]); in InferDevMatrixShape()
Doperator_info.cc641 Shape GetSliceShape(const Shape &tensor_shape, const Dimensions &strategy) { in GetSliceShape() function
685 Shape slice_shape = GetSliceShape(shapes.at(i), strategys.at(i)); in InferSliceShapeByStrategy()