Home
last modified time | relevance | path

Searched defs:IsSameShape (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/trt/
Dtrt_utils.h92 static bool IsSameShape(const nvinfer1::Dims &lhs, const nvinfer1::Dims &rhs) { in IsSameShape() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/common/
Dtensor_util.cc667 bool IsSameShape(const Tensor *input_1, const Tensor *input_2) { in IsSameShape() function
794 bool IsSameShape(const Tensor *input_1, const Tensor *input_2) { in IsSameShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dtensor_redistribution.cc888 bool IsSameShape(const Shape &src, const Shape &tgt) { in IsSameShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dcommon_utils.cc138 bool IsSameShape(const ShapeVector &shape_a, const ShapeVector &shape_b) { return shape_a == shape_… in IsSameShape() function