Searched defs:GetOutputShape (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/ |
| D | cache_manager.cc | 58 ShapeVector CacheManager::GetOutputShape(const AnfNodePtr &node, size_t index) { in GetOutputShape() function in mindspore::opt::CacheManager
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/ascend/model/ |
| D | model_infer.cc | 186 const std::vector<ShapeVector> ModelInfer::GetOutputShape() { return model_process_.GetOutputShape(… in GetOutputShape() function in mindspore::kernel::acl::ModelInfer
|
| D | model_process.cc | 271 const std::vector<ShapeVector> ModelProcess::GetOutputShape() { in GetOutputShape() function in mindspore::kernel::acl::ModelProcess
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | argmax_with_value.cc | 30 inline BaseShapePtr GetOutputShape(const ShapeVector &output_shape) { in GetOutputShape() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/adapter/ |
| D | callback_impl.cc | 83 ShapeVector CallbackImpl::GetOutputShape(const AnfNodePtr &node, size_t i) { in GetOutputShape() function in mindspore::graphkernel::CallbackImpl 374 ShapeVector CallbackImplWithInferShape::GetOutputShape(const AnfNodePtr &node, size_t i) { in GetOutputShape() function in mindspore::graphkernel::CallbackImplWithInferShape
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/ |
| D | callback_impl.cc | 34 ShapeVector CallbackImpl::GetOutputShape(const AnfNodePtr &node, size_t i) { return GetOutputInferS… in GetOutputShape() function in mindspore::graphkernel::CallbackImpl
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | matrix_diag_v3.cc | 55 ShapeVector GetOutputShape(const std::vector<int64_t> &x_shape, int64_t lower_diag_index, int64_t u… in GetOutputShape() function
|
| D | avg_pool_3d.cc | 80 std::vector<int64_t> GetOutputShape(const PrimitivePtr &primitive, const std::vector<int64_t> &in_s… in GetOutputShape() function
|
| D | max_pool3d.cc | 142 std::vector<int64_t> GetOutputShape(const PrimitivePtr &primitive, const std::vector<int64_t> &in_s… in GetOutputShape() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/ |
| D | anfalgo.cc | 732 ShapeVector GetOutputShape(const abstract::AbstractBasePtr &abstract, size_t output_idx, bool is_re… in GetOutputShape() function
|