Home
last modified time | relevance | path

Searched refs:ge_shape (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dutil.cc111 std::vector<int64_t> ge_shape; in GetGeTensorDesc() local
114 ge_shape.push_back(static_cast<int64_t>(me_shape[0])); in GetGeTensorDesc()
116 ge_shape.resize(me_shape.size()); in GetGeTensorDesc()
117 (void)std::transform(me_shape.begin(), me_shape.end(), ge_shape.begin(), IntegerCastFunc); in GetGeTensorDesc()
120 GeShape shape(ge_shape); in GetGeTensorDesc()
278 bool IsGeShapeCompatible(const GeShape &ge_shape, const ShapeVector &request_dims) { in IsGeShapeCompatible() argument
279 MS_LOG(INFO) << "GeTensor's shape is " << TransformUtil::PrintVector(ge_shape.GetDims()); in IsGeShapeCompatible()
283 std::vector<int64_t> ge_dims = ge_shape.GetDims(); in IsGeShapeCompatible()
321 ShapeVector TransformUtil::ConvertGeShape(const GeShape &ge_shape) { in ConvertGeShape() argument
323 std::vector<int64_t> ge_dims = ge_shape.GetDims(); in ConvertGeShape()
[all …]
Dutil.h162 static ShapeVector ConvertGeShape(const GeShape &ge_shape);
178 static ShapeVector ConvertGeShape(const GeShape &ge_shape, const ShapeVector &request_dims);
Dop_adapter_util.cc219 std::vector<int64_t> ge_shape; in ConvertAnyUtil() local
220 GeShape shape(ge_shape); in ConvertAnyUtil()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/tiling/
Dop_tiling_adapter.cc90 std::vector<int64_t> ge_shape; in ConvertInputShapeAndType() local
94 std::transform(ms_shape.begin(), ms_shape.end(), std::back_inserter(ge_shape), in ConvertInputShapeAndType()
103 ge_tensor_desc.SetShape(ge::GeShape(ge_shape)); in ConvertInputShapeAndType()
120 std::vector<int64_t> ge_shape; in ConvertOutputShapeAndType() local
124 std::transform(ms_shape.begin(), ms_shape.end(), std::back_inserter(ge_shape), in ConvertOutputShapeAndType()
132 ge_tensor_desc.SetShape(ge::GeShape(ge_shape)); in ConvertOutputShapeAndType()
160 std::vector<int64_t> ge_shape; in NewConstantOp() local
162 std::transform(ms_shape.begin(), ms_shape.end(), std::back_inserter(ge_shape), in NewConstantOp()
171 ge_tensor_desc.SetShape(ge::GeShape(ge_shape)); in NewConstantOp()
/third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/
Dconverter.cc176 std::vector<int64_t> ge_shape; in GenerateStubGeNode() local
178 std::transform(ms_shape.begin(), ms_shape.end(), std::back_inserter(ge_shape), in GenerateStubGeNode()
181 ge::GeTensorDesc(ge::GeShape(ge_shape), ge::Format::FORMAT_NCHW, in GenerateStubGeNode()
185 std::vector<int64_t> ge_shape; in GenerateStubGeNode() local
187 std::transform(ms_shape.begin(), ms_shape.end(), std::back_inserter(ge_shape), in GenerateStubGeNode()
190 ge::GeTensorDesc(ge::GeShape(ge_shape), ge::Format::FORMAT_NCHW, in GenerateStubGeNode()
/third_party/mindspore/tests/ut/cpp/transform/
Dconvert_test.cc828 GeShape ge_shape({10, 1, 1, 1}); in TEST_F() local
830 ASSERT_TRUE(TransformUtil::ConvertGeShape(ge_shape, request_dims) == request_dims); in TEST_F()
834 GeShape ge_shape({10, 15, 1, 1}); in TEST_F() local
836 ASSERT_TRUE(TransformUtil::ConvertGeShape(ge_shape, request_dims) == request_dims); in TEST_F()
840 GeShape ge_shape({10, 13, 18, 1}); in TEST_F() local
842 ASSERT_TRUE(TransformUtil::ConvertGeShape(ge_shape, request_dims) == request_dims); in TEST_F()
846 GeShape ge_shape({1, 10, 1, 1}); in TEST_F() local
848 ASSERT_TRUE(TransformUtil::ConvertGeShape(ge_shape, request_dims) == request_dims); in TEST_F()
852 GeShape ge_shape({10, 1, 1, 2}); in TEST_F() local
854 …ASSERT_TRUE(TransformUtil::ConvertGeShape(ge_shape, request_dims) == TransformUtil::ConvertGeShape… in TEST_F()
[all …]