| /external/tensorflow/tensorflow/core/kernels/ |
| svd_op_impl.h | 42 OP_REQUIRES_OK(context, context->GetAttr("full_matrices", &full_matrices_));
60 TensorShape({m, full_matrices_ ? m : min_size}),
61 TensorShape({n, full_matrices_ ? n : min_size})});
85 options = full_matrices_ ? Eigen::ComputeFullU | Eigen::ComputeFullV
98 bool full_matrices_; member in class:tensorflow::SvdOp
|
| qr_op_impl.h | 56 OP_REQUIRES_OK(context, context->GetAttr("full_matrices", &full_matrices_));
72 if (full_matrices_) {
105 if (full_matrices_) {
120 bool full_matrices_; member in class:tensorflow::QrOp
133 OP_REQUIRES_OK(context, context->GetAttr("full_matrices", &full_matrices_));
152 // If full_matrices_ is true then Q is m x m and R is m x n.
156 q_shape.set_dim(ndims - 1, full_matrices_ ? m : min_size);
161 r_shape.set_dim(ndims - 2, full_matrices_ ? m : min_size);
217 if (full_matrices_ || m == n) {
228 full_matrices_ ? m : min_size, &alpha
290 bool full_matrices_; member in class:tensorflow::QrOpGpu
[all...] |
| svd_op_gpu.cu.cc | 92 OP_REQUIRES_OK(context, context->GetAttr("full_matrices", &full_matrices_));
122 if (full_matrices_) {
162 cfg2D, m, full_matrices_ ? m : p, input_copy.flat<Scalar>().data(),
218 if (full_matrices_) {
288 if (full_matrices_) {
361 if (full_matrices_) {
404 bool full_matrices_; member in class:tensorflow::SvdOpGpu
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| svd_op_test.py | 56 for full_matrices_ in True, False:
61 matrix1, compute_uv=compute_uv_, full_matrices=full_matrices_)
63 matrix2, compute_uv=compute_uv_, full_matrices=full_matrices_)
67 matrix1, compute_uv=compute_uv_, full_matrices=full_matrices_)
69 matrix2, compute_uv=compute_uv_, full_matrices=full_matrices_)
83 full_matrices_):
105 def CheckApproximation(self, a, u, s, v, full_matrices_, tol):
111 if full_matrices_:
151 x_tf, compute_uv=compute_uv_, full_matrices=full_matrices_)
159 x_tf, compute_uv=compute_uv_, full_matrices=full_matrices_)
[all...] |
| qr_op_test.py | 55 for full_matrices_ in True, False:
60 q1, r1 = linalg_ops.qr(matrix1, full_matrices=full_matrices_)
61 q2, r2 = linalg_ops.qr(matrix2, full_matrices=full_matrices_)
70 def _GetQrOpTest(dtype_, shape_, full_matrices_, use_static_shape_):
129 q_tf, r_tf = linalg_ops.qr(x_tf, full_matrices=full_matrices_)
143 if full_matrices_:
161 def _GetQrGradOpTest(dtype_, shape_, full_matrices_):
178 tf_b = linalg_ops.qr(tf_a, full_matrices=full_matrices_)
|
