Home
last modified time | relevance | path

Searched defs:r_shape (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dapply_came_part4.cc83 …auto r_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->GetShape(… in CheckInputsShapePart4() local
145 ShapeVector r_shape = out_shape; in InferShape() local
Dqr.cc66 auto r_shape = std::make_shared<abstract::TensorShape>(out_r_dims); in InferShape() local
Dapply_came_part2.cc92 …auto r_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex4]->GetShape(… in CheckInputsShapePart2() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/
Dqr_scipy_cpu_kernel.cc52 auto r_shape = Convert2SizeTClipNeg(outputs[kIndex1]->GetShapeVector()); in Resize() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_qr.cc38 ShapeVector r_shape; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dlinalg_ops_proto.cc116 Shape r_shape; in CUST_IMPLEMT_INFERFUNC() local
426 Shape r_shape; in IMPLEMT_INFERFUNC() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/symbol_ops_impl/
Dshape_calc.cc201 auto r_shape = Emit(std::make_shared<Reduce>(inp, axis, keep_dims, skip_mode)); in Eval() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expanders/
Dlu.cc169 ShapeVector r_shape{num}; in Expand() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/grad_ops/
Dcommon_utils.cc509 auto r_shape = ReduceShape(x_shape, axis_value, skip_mode_); in Calc() local
Dgrad_array_ops.cc1904 auto r_shape = TileShape(inputs.at(1), inputs.at(0)); in __anond7f8849a1802() local