Home
last modified time | relevance | path

Searched defs:GetOutputShapes (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dsparse_slice.cc34 std::vector<abstract::BaseShapePtr> GetOutputShapes(const int64_t max_value_num, const int64_t rank… in GetOutputShapes() function
Dsparse_split.cc61 std::vector<abstract::BaseShapePtr> GetOutputShapes(const ShapeVector &output_indices_vec, in GetOutputShapes() function
Dsspaddmm.cc385 std::vector<abstract::BaseShapePtr> GetOutputShapes(const PrimitivePtr &primitive, in GetOutputShapes() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/consumers/
Dpull_based_tree_consumer.cc139 Status TreeGetters::GetOutputShapes(std::vector<TensorShape> *shapes, bool estimate) { in GetOutputShapes() function in mindspore::dataset::TreeGetters
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/
Ddatasets.cc343 std::vector<std::vector<int64_t>> Dataset::GetOutputShapes() { in GetOutputShapes() function in mindspore::dataset::Dataset