Searched refs:CheckShapeSame (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | relu_grad.cc | 40 (void)abstract::CheckShapeSame(prim_name, out, dout); in InferShape()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | param_validator.h | 44 ShapePtr CheckShapeSame(const std::string &op, const AbstractTensorPtr &tensor_base, const Abstract…
|
| D | param_validator.cc | 113 ShapePtr CheckShapeSame(const std::string &op, const AbstractTensorPtr &tensor_base, const Abstract… in CheckShapeSame() function
|
| D | prim_maths.cc | 63 (void)CheckShapeSame(op_name, out, dout); in InferImplSqrtGrad()
|
| D | prim_arrays.cc | 117 (void)CheckShapeSame(op_name, tensor_base, tensor); in InferImplStack()
|