Home
last modified time | relevance | path

Searched refs:full_matrices (Results 1 – 25 of 39) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/tests/
Dqr_op_test.py84 def _test(self, x_np, full_matrices, full_rank=True): argument
90 q_tf, r_tf = linalg_ops.qr(x_tf, full_matrices=full_matrices)
100 if full_matrices:
119 for full_matrices in [True, False]:
123 self._test(x_np, full_matrices)
127 self._test(x_np, full_matrices=True)
132 self._test(x_np, full_matrices=True)
138 self._test(x_np, full_matrices=True)
144 self._test(x_np, full_matrices=True, full_rank=False)
Dsvd_op_test.py50 s, u, v = linalg_ops.svd(x_tf, full_matrices=True)
71 x_tf, full_matrices=True, compute_uv=False)
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dqr_op_test.py70 q1, r1 = linalg_ops.qr(matrix1, full_matrices=full_matrices_)
71 q2, r2 = linalg_ops.qr(matrix2, full_matrices=full_matrices_)
139 q_tf, r_tf = linalg_ops.qr(x_tf, full_matrices=full_matrices_)
180 ret = linalg_ops.qr(x, full_matrices=True)
212 lambda a: linalg_ops.qr(a, full_matrices=full_matrices_)[0],
213 lambda a: linalg_ops.qr(a, full_matrices=full_matrices_)[1]
281 for full_matrices in False, True:
288 full_matrices,
291 _GetQrOpTest(dtype, shape, full_matrices,
296 for full_matrices in False, True:
[all …]
Dsvd_op_test.py97 matrix1, compute_uv=compute_uv_, full_matrices=full_matrices_)
99 matrix2, compute_uv=compute_uv_, full_matrices=full_matrices_)
103 matrix1, compute_uv=compute_uv_, full_matrices=full_matrices_)
105 matrix2, compute_uv=compute_uv_, full_matrices=full_matrices_)
192 x_tf, compute_uv=compute_uv_, full_matrices=full_matrices_)
201 x_tf, compute_uv=compute_uv_, full_matrices=full_matrices_)
210 x_np, compute_uv=compute_uv_, full_matrices=full_matrices_)
213 x_np, compute_uv=compute_uv_, full_matrices=full_matrices_)
237 tf_a, compute_uv=True, full_matrices=full_matrices_)
402 for full_matrices in False, True:
[all …]
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_Qr.pbtxt13 Orthonormal basis for range of `a`. If `full_matrices` is `False` then
14 shape is `[..., M, P]`; if `full_matrices` is `True` then shape is
21 Triangular factor. If `full_matrices` is `False` then shape is
22 `[..., P, N]`. If `full_matrices` is `True` then shape is `[..., M, N]`.
26 name: "full_matrices"
46 q_full, r_full = qr(a, full_matrices=True)
Dapi_def_Svd.pbtxt19 Left singular vectors. If `full_matrices` is `False` then shape is
20 `[..., M, P]`; if `full_matrices` is `True` then shape is
27 Left singular vectors. If `full_matrices` is `False` then shape is
28 `[..., N, P]`. If `full_matrices` is `True` then shape is `[..., N, N]`.
41 name: "full_matrices"
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dqr_test.cc49 for (bool full_matrices : {false, true}) { in XLA_TEST_F()
56 xla::QrExplicit(a, full_matrices, q, r); in XLA_TEST_F()
70 xla::F32, {m, full_matrices ? m : std::min(m, n)})); in XLA_TEST_F()
73 xla::F32, {full_matrices ? m : std::min(m, n), n})); in XLA_TEST_F()
Dqr.cc110 void QrExplicit(XlaOp a, bool full_matrices, XlaOp& q, XlaOp& r) { in QrExplicit() argument
123 if (full_matrices) { in QrExplicit()
Dqr.h48 void QrExplicit(XlaOp a, bool full_matrices, XlaOp& q, XlaOp& r);
/external/tensorflow/tensorflow/core/ops/
Dlinalg_ops_test.cc202 auto set_attrs = [&op](bool full_matrices) { in TEST() argument
206 .Attr("full_matrices", full_matrices) in TEST()
212 .Attr("full_matrices", full_matrices) in TEST()
253 auto set_attrs = [&op](bool compute_uv, bool full_matrices) { in TEST() argument
258 .Attr("full_matrices", full_matrices) in TEST()
265 .Attr("full_matrices", full_matrices) in TEST()
Dlinalg_ops.cc223 bool full_matrices; in QrShapeFn() local
224 TF_RETURN_IF_ERROR(c->GetAttr("full_matrices", &full_matrices)); in QrShapeFn()
225 if (full_matrices) { in QrShapeFn()
260 bool full_matrices; in SvdShapeFn() local
261 TF_RETURN_IF_ERROR(c->GetAttr("full_matrices", &full_matrices)); in SvdShapeFn()
262 if (full_matrices) { in SvdShapeFn()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DQr.pbtxt16 name: "full_matrices"
50 name: "full_matrices"
DSvd.pbtxt27 name: "full_matrices"
72 name: "full_matrices"
DBatchSvd.pbtxt27 name: "full_matrices"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DQr.pbtxt16 name: "full_matrices"
50 name: "full_matrices"
DSvd.pbtxt27 name: "full_matrices"
72 name: "full_matrices"
DBatchSvd.pbtxt27 name: "full_matrices"
/external/tensorflow/tensorflow/python/ops/
Dlinalg_ops.py488 def svd(tensor, full_matrices=False, compute_uv=True, name=None): argument
549 tensor, compute_uv=compute_uv, full_matrices=full_matrices, name=name)
Dlinalg_grad.py837 full_matrices = op.get_attr("full_matrices")
870 if full_matrices and abs(m - n) > 1:
915 if full_matrices:
Dinit_ops.py953 q, r = gen_linalg_ops.qr(a, full_matrices=False)
1013 q, r = gen_linalg_ops.qr(a, full_matrices=False)
Dinit_ops_v2.py695 q, r = gen_linalg_ops.qr(a, full_matrices=False)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.linalg.pbtxt229 …argspec: "args=[\'input\', \'full_matrices\', \'name\'], varargs=None, keywords=None, defaults=[\'…
249 …argspec: "args=[\'tensor\', \'full_matrices\', \'compute_uv\', \'name\'], varargs=None, keywords=N…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.linalg.pbtxt217 …argspec: "args=[\'input\', \'full_matrices\', \'name\'], varargs=None, keywords=None, defaults=[\'…
237 …argspec: "args=[\'tensor\', \'full_matrices\', \'compute_uv\', \'name\'], varargs=None, keywords=N…
/external/tensorflow/tensorflow/tools/compatibility/testdata/
Dtest_file_v0_11.py169 tf.svd(mat, compute_uv=False, full_matrices=True).eval())
/external/tensorflow/tensorflow/compiler/xla/python/
Dops.cc287 [](XlaOp a, bool full_matrices) -> StatusOr<std::pair<XlaOp, XlaOp>> { in BuildOpsSubmodule() argument
289 QrExplicit(a, full_matrices, q, r); in BuildOpsSubmodule()

12