Searched defs:compute_uv (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | svd.cc | 51 auto compute_uv = GetValue<bool>(prim->GetAttr(kAttrComputeUV)); in SvdInferShape() local
108 void Svd::Init(const bool full_matrices, const bool compute_uv) { in Init()
117 void Svd::set_compute_uv(const bool compute_uv) { (void)this->AddAttr(kAttrComputeUV, api::MakeValu… in set_compute_uv()
120 bool Svd::compute_uv() const { return GetValue<bool>(GetAttr(kAttrComputeUV)); } in compute_uv() function in mindspore::ops::Svd
|
| /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/python/mindspore/common/ |
| D | tensor.py | 3548 def svd(self, full_matrices=False, compute_uv=True): argument
|