Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dminimum_grad_cpu_kernel.cc55 void CheckShape(std::vector<size_t> *shape) { in CheckShape() function
Dmaximum_grad_cpu_kernel.cc27 void CheckShape(std::vector<size_t> *shape) { in CheckShape() function
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/model/
Dop_node.h65 virtual void CheckShape(const NodePtrList &inputs, const DAttrs &attrs) {} in CheckShape() function
/third_party/mindspore/mindspore/core/abstract/
Dprim_nn.cc234 void CheckShape(const std::string &op_name, const ShapeVector &w_shape, const AbstractTensorPtr &in… in CheckShape() function