| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | batch_matmul_op_test.py | 49 def _npBatchMatmul(self, x, y, adjoint_a, adjoint_b): argument
51 if adjoint_a:
58 def _compare(self, x_in, y_in, adjoint_a, adjoint_b, static_shape): argument
61 x = x_in if not adjoint_a else x_in.reshape(x_t_shape)
67 z0 = math_ops.matmul(x, y, adjoint_a=adjoint_a, adjoint_b=adjoint_b)
73 x_ph, y_ph, adjoint_a=adjoint_a, adjoint_b=adjoint_b)
75 z1 = self._npBatchMatmul(x, y, adjoint_a, adjoint_b)
78 def _testNonEmpty(self, dtype, adjoint_a, adjoint_b, use_static_shape): argument
84 adjoint_a,
98 def _testBroadcasting(self, dtype, adjoint_a, adjoint_b, use_static_shape): argument
[all …]
|
| D | sparse_tensor_dense_matmul_op_test.py | 53 adjoint_a=False, argument
57 if adjoint_a:
73 sp_x_value, y, adjoint_a=adjoint_a, adjoint_b=adjoint_b)
77 adjoint_a=adjoint_a,
159 sparse_t, dense_t, adjoint_a=True))
164 sparse_t, dense_t, adjoint_a=True))
194 sparse_t, dense_t, adjoint_a=True))
201 sparse_t, dense_t, adjoint_a=True))
217 self._testMatmul(x, y, adjoint_a=False, adjoint_b=False)
218 self._testMatmul(x.transpose(), y, adjoint_a=True, adjoint_b=False)
[all …]
|
| D | sparse_tensor_dense_matmul_grad_test.py | 62 def _testGradients(self, adjoint_a, adjoint_b, name, values_dtype, argument
68 adjoint=adjoint_a,
74 sp_t, dense_t, adjoint_a=adjoint_a, adjoint_b=adjoint_b, name=name)
86 for adjoint_a in [True, False]:
89 adjoint_a, adjoint_b, values_dtype.__name__, indices_dtype.__name__)
90 self._testGradients(adjoint_a, adjoint_b, name, values_dtype,
|
| D | matmul_op_test.py | 257 for adjoint_a, transpose_a in trans_options:
260 use_static_shape, dtype.__name__, m, n, k, adjoint_a,
267 adjoint_a=adjoint_a,
276 adjoint_a=adjoint_a,
|
| D | cholesky_op_test.py | 44 middle = math_ops.matmul(l, grad, adjoint_a=True)
49 math_ops.matmul(l_inverse, middle, adjoint_a=True), l_inverse)
58 middle = math_ops.matmul(l, grad, adjoint_a=True)
189 matrix1 = math_ops.matmul(matrix1, matrix1, adjoint_a=True)
190 matrix2 = math_ops.matmul(matrix2, matrix2, adjoint_a=True)
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| D | csr_sparse_matrix_test.py | 151 def _testSparseSparse(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
155 dense_shape_a = [5, 13, 7] if transpose_a or adjoint_a else [5, 7, 13]
170 adjoint_a=adjoint_a,
177 adjoint_a=adjoint_a,
191 def _testSparseDense(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
196 dense_shape_a = [5, 13, 7] if transpose_a or adjoint_a else [5, 7, 13]
210 adjoint_a=adjoint_a,
217 adjoint_a=adjoint_a,
229 def _testDenseSparse(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
234 dense_shape_a = [5, 13, 7] if transpose_a or adjoint_a else [5, 7, 13]
[all …]
|
| D | csr_sparse_matrix_dense_mat_mul_grad_test.py | 62 transpose_b, adjoint_a, adjoint_b, argument
71 if transpose_a or adjoint_a:
73 if adjoint_a:
90 adjoint_a=adjoint_a,
|
| D | csr_sparse_matrix_sparse_mat_mul_grad_test.py | 62 transpose_b, adjoint_a, argument
71 if transpose_a or adjoint_a:
73 if adjoint_a:
92 adjoint_a=adjoint_a,
|
| D | csr_sparse_matrix_ops_test.py | 547 for (adjoint_a, adjoint_b) in ((False, False), (False, True),
549 if (transpose_a and adjoint_a) or (transpose_b and adjoint_b):
556 if transpose_a or adjoint_a:
568 (transpose_a, transpose_b, adjoint_a, adjoint_b))
577 adjoint_a=adjoint_a,
584 adjoint_a=adjoint_a,
609 for (adjoint_a, adjoint_b) in ((False, False), (False, True),
611 if (transpose_a and adjoint_a) or (transpose_b and adjoint_b):
618 if transpose_a or adjoint_a:
630 (transpose_a, transpose_b, adjoint_a, adjoint_b))
[all …]
|
| /external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
| D | sparse_csr_matrix_ops.py | 121 def matmul_shape_inference(a, b, c, transpose_a, transpose_b, adjoint_a, argument
130 transpose_a = transpose_a or adjoint_a
151 adjoint_a=False, argument
187 adjoint_a=adjoint_a,
204 adjoint_a=adjoint_a,
212 transpose_b, adjoint_a, adjoint_b)
221 adjoint_a=adjoint_a,
225 if not adjoint_a and not adjoint_b:
236 adjoint_a=not adjoint_b,
237 adjoint_b=not adjoint_a,
|
| D | sparse_csr_matrix_grad.py | 123 grad_a_dense = matmul(b, grad, adjoint_a=adj_b, adjoint_b=True)
127 adjoint_a=not adj_a,
152 grad_a_dense = matmul(b, conj(grad), adjoint_a=adj_b)
156 adjoint_a=not adj_a,
194 grad_b = matmul(a, grad, adjoint_a=True, type=dtype)
197 grad_b = matmul(grad, a, adjoint_a=True, type=dtype)
203 grad_a = matmul(b, grad, adjoint_a=True, adjoint_b=True, type=dtype)
204 grad_b = matmul(grad, a, adjoint_a=True, adjoint_b=True, type=dtype)
213 grad_a = matmul(b, grad, adjoint_a=True, transpose_b=True, type=dtype)
220 grad_b = matmul(grad, a, adjoint_a=True, transpose_b=True, type=dtype)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_tensor_dense_matmul_op_test.cc | 27 Node* a_shape, Node* b, bool adjoint_a, in SparseTensorDenseMatMulNode() argument
36 .Attr("adjoint_a", adjoint_a) in SparseTensorDenseMatMulNode()
43 bool adjoint_a, bool adjoint_b) { in SparseTensorDenseMatmul() argument
49 a_shape_t(0) = adjoint_a ? k : m; in SparseTensorDenseMatmul()
50 a_shape_t(1) = adjoint_a ? m : k; in SparseTensorDenseMatmul()
67 test::graph::Constant(g, b), adjoint_a, adjoint_b); in SparseTensorDenseMatmul()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_grad.py | 55 ainv, math_ops.matmul(grad, ainv, adjoint_b=True), adjoint_a=True)
475 middle = math_ops.matmul(l, grad, adjoint_a=True)
481 math_ops.matmul(l_inverse, middle, adjoint_a=True), l_inverse)
513 qdq = math_ops.matmul(q, dq, adjoint_a=True)
555 adjoint_a = op.get_attr("adjoint")
557 grad_b = linalg_ops.matrix_solve(a, grad, adjoint=not adjoint_a)
558 if adjoint_a:
619 a1 = math_ops.matmul(tmp, a, adjoint_a=True)
621 a2 = grad - math_ops.matmul(a, grad_b, adjoint_a=True)
650 adjoint_a = op.get_attr("adjoint")
[all …]
|
| D | sparse_ops_test.py | 163 independent_dense_tf, sp, adjoint_a=False, adjoint_b=False)
168 independent_dense_tf, sp, adjoint_a=False, adjoint_b=True)
174 independent_dense_tf, sp, adjoint_a=True, adjoint_b=False)
180 independent_dense_tf, sp, adjoint_a=True, adjoint_b=True)
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseTensorDenseMatMul.pbtxt | 28 name: "adjoint_a"
46 if adjoint_a == false:
49 if adjoint_a == true:
|
| D | api_def_SparseMatrixSparseMatMul.pbtxt | 25 name: "adjoint_a"
39 according to the Boolean parameters `transpose_a`, `adjoint_a`, `transpose_b`
40 and `adjoint_b`. At most one of `transpose_a` or `adjoint_a` may be True.
108 adjoint_a: If True, `a` adjointed before multiplication.
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | SparseTensorDenseMatMul.pbtxt | 28 name: "adjoint_a"
82 name: "adjoint_a"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SparseTensorDenseMatMul.pbtxt | 28 name: "adjoint_a"
82 name: "adjoint_a"
|
| D | SparseMatrixSparseMatMul.pbtxt | 42 name: "adjoint_a"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_csr_matrix_ops.cc | 254 bool adjoint_a = false; in __anon59a9d6c30702() local
256 TF_RETURN_IF_ERROR(c->GetAttr("adjoint_a", &adjoint_a)); in __anon59a9d6c30702()
258 if (adjoint_a && transpose_a) { in __anon59a9d6c30702()
266 transpose_a = transpose_a || adjoint_a; in __anon59a9d6c30702()
412 bool adjoint_a = false; in __anon59a9d6c30a02() local
414 TF_RETURN_IF_ERROR(c->GetAttr("adjoint_a", &adjoint_a)); in __anon59a9d6c30a02()
416 if (adjoint_a && transpose_a) { in __anon59a9d6c30a02()
423 transpose_a = transpose_a || adjoint_a; in __anon59a9d6c30a02()
|
| D | sparse_csr_matrix_ops_test.cc | 108 auto set_options = [&op](bool transpose_a, bool transpose_b, bool adjoint_a, in TEST()
115 .Attr("adjoint_a", adjoint_a) in TEST()
232 auto set_options = [&op](bool transpose_a, bool transpose_b, bool adjoint_a, in TEST()
239 .Attr("adjoint_a", adjoint_a) in TEST()
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_low_rank_update.py | 373 uh_x = math_ops.matmul(u, x, adjoint_a=True, adjoint_b=adjoint_arg)
378 vh_x = math_ops.matmul(v, x, adjoint_a=True, adjoint_b=adjoint_arg)
442 vh_linv_rhs = math_ops.matmul(v, linv_rhs, adjoint_a=True)
465 vh_linv_u = math_ops.matmul(self.v, linv_u, adjoint_a=True)
|
| D | linear_operator_util.py | 375 matrix, rhs, adjoint_a=adjoint)
390 adjoint_a=False, argument
429 a_domain_sz_ = a.shape[-2 if adjoint_a or transpose_a else -1]
442 if adjoint_a:
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_mat_mul_op.cc | 298 bool adjoint_a; in CSRSparseMatMulGPUOp() local
299 OP_REQUIRES_OK(c, c->GetAttr("adjoint_a", &adjoint_a)); in CSRSparseMatMulGPUOp()
300 OP_REQUIRES(c, !(adjoint_a && transpose_a_), in CSRSparseMatMulGPUOp()
308 conjugate_a_ = adjoint_a; in CSRSparseMatMulGPUOp()
310 transpose_a_ = transpose_a_ || adjoint_a; in CSRSparseMatMulGPUOp()
555 bool adjoint_a, bool transpose_b) in CSRSparseSparseMatrixMatMul()
560 adjoint_a_(adjoint_a), in CSRSparseSparseMatrixMatMul()
569 ? (adjoint_a ? GPUSPARSE(OPERATION_TRANSPOSE) in CSRSparseSparseMatrixMatMul()
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | linear_operator_test.py | 108 self._matrix, x, adjoint_a=adjoint, adjoint_b=adjoint_arg)
386 operator, rhs, adjoint_a=adjoint, adjoint_b=adjoint_arg)
408 lhs, mat, adjoint_a=adjoint, adjoint_b=adjoint_arg)
410 lhs, right_operator, adjoint_a=adjoint, adjoint_b=adjoint_arg)
|