Home
last modified time | relevance | path

Searched defs:compute_uv (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dsvd.cc51 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/
Dlinalg_ops.py72 def __init__(self, full_matrices=False, compute_uv=True): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/
Dtest_svd_op.py43 def __init__(self, full_matrices=False, compute_uv=True): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/
Dtest_svd_op.py44 def __init__(self, full_matrices=False, compute_uv=True): argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/
Dlinalg_func.py184 def svd(input, full_matrices=False, compute_uv=True): argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/
Dtensor.py3548 def svd(self, full_matrices=False, compute_uv=True): argument