Searched defs:full_matrices (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/st/dyn_shape_dev/ |
| D | test_qr_dyn_shape.py | 24 def qr_forward_func(x, full_matrices): argument
29 def qr_backward_func(x, full_matrices): argument
74 def qr_(x, full_matrices): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | svd.cc | 52 auto full_matrices = GetValue<bool>(prim->GetAttr(kAttrFullMatrices)); in SvdInferShape() local
108 void Svd::Init(const bool full_matrices, const bool compute_uv) { in Init()
113 void Svd::set_full_matrices(const bool full_matrices) { in set_full_matrices()
119 bool Svd::full_matrices() const { return GetValue<bool>(GetAttr(kAttrFullMatrices)); } in full_matrices() function in mindspore::ops::Svd
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_qr.cc | 36 ValuePtr full_matrices; member
60 auto full_matrices = shape_param.full_matrices->ToAbstract(); in TEST_P() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | linalg_ops.py | 72 def __init__(self, full_matrices=False, compute_uv=True): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_svd_op.py | 43 def __init__(self, full_matrices=False, compute_uv=True): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_svd_op.py | 44 def __init__(self, full_matrices=False, compute_uv=True): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | linalg_func.py | 184 def svd(input, full_matrices=False, compute_uv=True): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | qr.cc | 87 bool full_matrices = attr_full_matrices->GetBool(); in QrCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | linalg_ops_proto.cc | 427 auto full_matrices = op.get_attr_full_matrices(); in IMPLEMT_INFERFUNC() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/ |
| D | gen_lite_ops.cc | 1517 void Qr::set_full_matrices(const bool &full_matrices) { (void)this->AddAttr("full_matri… in set_full_matrices()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/ |
| D | tensor.py | 3548 def svd(self, full_matrices=False, compute_uv=True): argument
|