Home
last modified time | relevance | path

Searched defs:MatrixSolve (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dmatrix_solve.h37 MatrixSolve() : BaseOperator(kNameMatrixSolve) { InitIOName({"matrix", "rhs"}, {"output"}); } in MatrixSolve() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/utils/
Dlinalg_ops_shape_fns.cc57 graphStatus MatrixSolve(const TensorDesc &tensor1, const TensorDesc &tensor2, bool square, Shape &o… in MatrixSolve() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dlinalg_ops_proto.cc27 IMPLEMT_INFERFUNC(MatrixSolve, MatrixSolveInfer) { in IMPLEMT_INFERFUNC() argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Dmath_ops.py3884 class MatrixSolve(Primitive): class
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_info_cfg/
Dcust_aicpu_kernel.ini2037 [MatrixSolve] section
2056 [MatrixSolveLs] argument