Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/tests/
Dqr_op_test.py88 def _test(self, x_np, full_matrices, full_rank=True): argument
94 q_tf, r_tf = linalg_ops.qr(x_tf, full_matrices=full_matrices)
104 if full_matrices:
123 for full_matrices in [True, False]:
127 self._test(x_np, full_matrices)
131 self._test(x_np, full_matrices=True)
136 self._test(x_np, full_matrices=True)
142 self._test(x_np, full_matrices=True)
148 self._test(x_np, full_matrices=True, full_rank=False)
Dsvd_op_test.py54 s, u, v = linalg_ops.svd(x_tf, full_matrices=True)
75 x_tf, full_matrices=True, compute_uv=False)
/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/python/kernel_tests/
Dqr_op_test.py74 q1, r1 = linalg_ops.qr(matrix1, full_matrices=full_matrices_)
75 q2, r2 = linalg_ops.qr(matrix2, full_matrices=full_matrices_)
143 q_tf, r_tf = linalg_ops.qr(x_tf, full_matrices=full_matrices_)
184 ret = linalg_ops.qr(x, full_matrices=True)
216 lambda a: linalg_ops.qr(a, full_matrices=full_matrices_)[0],
217 lambda a: linalg_ops.qr(a, full_matrices=full_matrices_)[1]
285 for full_matrices in False, True:
292 full_matrices,
295 _GetQrOpTest(dtype, shape, full_matrices,
300 for full_matrices in False, True:
[all …]
Dsvd_op_test.py75 matrix1, compute_uv=compute_uv_, full_matrices=full_matrices_)
77 matrix2, compute_uv=compute_uv_, full_matrices=full_matrices_)
81 matrix1, compute_uv=compute_uv_, full_matrices=full_matrices_)
83 matrix2, compute_uv=compute_uv_, full_matrices=full_matrices_)
162 x_tf, compute_uv=compute_uv_, full_matrices=full_matrices_)
171 x_tf, compute_uv=compute_uv_, full_matrices=full_matrices_)
180 x_np, compute_uv=compute_uv_, full_matrices=full_matrices_)
183 x_np, compute_uv=compute_uv_, full_matrices=full_matrices_)
207 tf_a, compute_uv=True, full_matrices=full_matrices_)
372 for full_matrices in False, True:
[all …]
/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/compiler/xla/client/lib/
Dqr.cc56 XlaOp a, bool full_matrices, int64 block_size, in QRDecomposition() argument
84 if (!full_matrices) { in QRDecomposition()
Dqr.h37 XlaOp a, bool full_matrices, int64 block_size = 128,
/external/tensorflow/tensorflow/python/ops/
Dlinalg_ops.py491 def svd(tensor, full_matrices=False, compute_uv=True, name=None): argument
552 tensor, compute_uv=compute_uv, full_matrices=full_matrices, name=name)
Dlinalg_grad.py833 full_matrices = op.get_attr("full_matrices")
866 if full_matrices and abs(m - n) > 1:
910 if full_matrices:
Dinit_ops_v2.py688 q, r = gen_linalg_ops.qr(a, full_matrices=False)
Dinit_ops.py589 q, r = gen_linalg_ops.qr(a, full_matrices=False)
646 q, r = gen_linalg_ops.qr(a, full_matrices=False)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.linalg.pbtxt213 …argspec: "args=[\'input\', \'full_matrices\', \'name\'], varargs=None, keywords=None, defaults=[\'…
233 …argspec: "args=[\'tensor\', \'full_matrices\', \'compute_uv\', \'name\'], varargs=None, keywords=N…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.linalg.pbtxt225 …argspec: "args=[\'input\', \'full_matrices\', \'name\'], varargs=None, keywords=None, defaults=[\'…
245 …argspec: "args=[\'tensor\', \'full_matrices\', \'compute_uv\', \'name\'], varargs=None, keywords=N…
/external/tensorflow/tensorflow/compiler/xla/python/
Dops.cc194 [](XlaOp a, bool full_matrices) -> StatusOr<std::pair<XlaOp, XlaOp>> { in BuildOpsSubmodule() argument
195 TF_ASSIGN_OR_RETURN(auto qr, QRDecomposition(a, full_matrices)); in BuildOpsSubmodule()
/external/tensorflow/tensorflow/tools/compatibility/testdata/
Dtest_file_v0_11.py172 tf.svd(mat, compute_uv=False, full_matrices=True).eval())
/external/tensorflow/tensorflow/python/keras/initializers/
Dinitializers_v2.py610 q, r = gen_linalg_ops.qr(a, full_matrices=False)

12