Home
last modified time | relevance | path

Searched defs:GetShapeSize (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/common/
Dtensor_util.cc241 size_t GetShapeSize(const TensorT &tensor) { in GetShapeSize() function
285 size_t GetShapeSize(const std::vector<int32_t> &shape) { in GetShapeSize() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Ddynamic_stitch_cpu_kernel.cc25 int64_t GetShapeSize(const ShapeVector &shape) { in GetShapeSize() function
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Darange.cc93 int64_t GetShapeSize(const TypePtr dtype, const ValuePtrList &input_values, bool result_type_is_int… in GetShapeSize() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dframework_utils.cc766 bool GetShapeSize(const ShapeVector &shape, const TypePtr &type_ptr, int64_t *size_i) { in GetShapeSize() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/
Deinsum_helper.h75 size_t GetShapeSize(const std::vector<int64_t> &shape) { in GetShapeSize() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tf/
Dtf_model_parser.cc109 int GetShapeSize(const tensorflow::TensorProto &tensor_proto) { in GetShapeSize() function