/external/tensorflow/tensorflow/python/kernel_tests/ |
D | batch_matmul_op_test.py | 49 def _npBatchMatmul(self, x, y, adjoint_a, adjoint_b): argument 53 if adjoint_b: 58 def _compare(self, x_in, y_in, adjoint_a, adjoint_b, static_shape): argument 62 y = y_in if not adjoint_b else y_in.reshape(y_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 85 adjoint_b, 98 def _testBroadcasting(self, dtype, adjoint_a, adjoint_b, use_static_shape): argument [all …]
|
D | sparse_tensor_dense_matmul_op_test.py | 54 adjoint_b=False, argument 60 if adjoint_b: 73 sp_x_value, y, adjoint_a=adjoint_a, adjoint_b=adjoint_b) 78 adjoint_b=adjoint_b) 217 self._testMatmul(x, y, adjoint_a=False, adjoint_b=False) 218 self._testMatmul(x.transpose(), y, adjoint_a=True, adjoint_b=False) 219 self._testMatmul(x, y.transpose(), adjoint_a=False, adjoint_b=True) 221 x.transpose(), y.transpose(), adjoint_a=True, adjoint_b=True) 234 for adjoint_b in [True, False]: 241 y = y.transpose() if adjoint_b else y [all …]
|
D | sparse_tensor_dense_matmul_grad_test.py | 62 def _testGradients(self, adjoint_a, adjoint_b, name, values_dtype, argument 71 dense_t = self._randomTensor([k, m], values_dtype, adjoint=adjoint_b) 74 sp_t, dense_t, adjoint_a=adjoint_a, adjoint_b=adjoint_b, name=name) 77 dense_t_shape = [m, k] if adjoint_b else [k, m] 87 for adjoint_b 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 | 258 for adjoint_b, transpose_b in trans_options: 261 transpose_a, adjoint_b, transpose_b) 269 adjoint_b=adjoint_b, 278 adjoint_b=adjoint_b,
|
/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 156 dense_shape_b = [5, 15, 13] if transpose_b or adjoint_b else [5, 13, 15] 171 adjoint_b=adjoint_b) 178 adjoint_b=adjoint_b) 191 def _testSparseDense(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument 197 dense_shape_b = [5, 15, 13] if transpose_b or adjoint_b else [5, 13, 15] 211 adjoint_b=adjoint_b) 218 adjoint_b=adjoint_b) 229 def _testDenseSparse(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument 235 dense_shape_b = [5, 15, 13] if transpose_b or adjoint_b else [5, 13, 15] [all …]
|
D | csr_sparse_matrix_dense_mat_mul_grad_test.py | 62 transpose_b, adjoint_a, adjoint_b, argument 77 if transpose_b or adjoint_b: 79 if adjoint_b: 91 adjoint_b=adjoint_b,
|
D | csr_sparse_matrix_sparse_mat_mul_grad_test.py | 63 adjoint_b): argument 78 if transpose_b or adjoint_b: 80 if adjoint_b: 93 adjoint_b=adjoint_b,
|
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): 558 if transpose_b or adjoint_b: 568 (transpose_a, transpose_b, adjoint_a, adjoint_b)) 578 adjoint_b=adjoint_b) 585 adjoint_b=adjoint_b) 609 for (adjoint_a, adjoint_b) in ((False, False), (False, True), 611 if (transpose_a and adjoint_a) or (transpose_b and adjoint_b): 620 if transpose_b or adjoint_b: 630 (transpose_a, transpose_b, adjoint_a, adjoint_b)) [all …]
|
/external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
D | sparse_csr_matrix_ops.py | 122 adjoint_b): argument 131 transpose_b = transpose_b or adjoint_b 152 adjoint_b=False, argument 188 adjoint_b=adjoint_b, 205 adjoint_b=adjoint_b, 212 transpose_b, adjoint_a, adjoint_b) 222 adjoint_b=adjoint_b) 225 if not adjoint_a and not adjoint_b: 236 adjoint_a=not adjoint_b, 237 adjoint_b=not adjoint_a, [all …]
|
D | sparse_csr_matrix_grad.py | 121 grad_a_dense = matmul(grad, b, adjoint_b=not adj_b) 123 grad_a_dense = matmul(b, grad, adjoint_a=adj_b, adjoint_b=True) 131 grad_a_dense = matmul(b, grad, transpose_a=True, adjoint_b=True) 150 grad_a_dense = matmul(grad, b, transpose_a=True, adjoint_b=not adj_b) 165 grad_b = sparse_matmul(a, grad, adjoint_b=True, transpose_output=True) 193 grad_a = matmul(grad, b, adjoint_b=True, type=dtype) 200 grad_a = matmul(b, grad, adjoint_b=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) 214 grad_b = matmul(grad, a, transpose_a=True, adjoint_b=True, type=dtype) [all …]
|
/external/tensorflow/tensorflow/core/kernels/ |
D | sparse_tensor_dense_matmul_op_test.cc | 28 bool adjoint_b) { in SparseTensorDenseMatMulNode() argument 37 .Attr("adjoint_b", adjoint_b) in SparseTensorDenseMatMulNode() 43 bool adjoint_a, bool adjoint_b) { in SparseTensorDenseMatmul() argument 61 Tensor b(DT_FLOAT, adjoint_b ? TensorShape({n, k}) : TensorShape({k, n})); 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) 515 rdr = math_ops.matmul(r, dr, adjoint_b=True) 546 dq + math_ops.matmul(y, dv, adjoint_b=True), 559 grad_a = -math_ops.matmul(c, grad_b, adjoint_b=True) 561 grad_a = -math_ops.matmul(grad_b, c, adjoint_b=True) 594 xzt = math_ops.matmul(x, z, adjoint_b=True) 596 grad_a = -math_ops.matmul(a, zx_sym) + math_ops.matmul(b, z, adjoint_b=True) 622 a2 = math_ops.matmul(tmp, a2, adjoint_b=True) 656 grad_a = -math_ops.matmul(c, grad_b, adjoint_b=True) 658 grad_a = -math_ops.matmul(grad_b, c, adjoint_b=True) [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_SparseMatrixSparseMatMul.pbtxt | 29 name: "adjoint_b" 40 and `adjoint_b`. At most one of `transpose_a` or `adjoint_a` may be True. 41 Similarly, at most one of `transpose_b` or `adjoint_b` may be True. 109 adjoint_b: If True, `b` adjointed before multiplication.
|
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
D | SparseTensorDenseMatMul.pbtxt | 35 name: "adjoint_b" 89 name: "adjoint_b"
|
D | SparseMatrixSparseMatMul.pbtxt | 49 name: "adjoint_b"
|
D | SparseMatrixMatMul.pbtxt | 41 name: "adjoint_b"
|
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
D | SparseTensorDenseMatMul.pbtxt | 35 name: "adjoint_b" 89 name: "adjoint_b"
|
D | SparseMatrixSparseMatMul.pbtxt | 49 name: "adjoint_b"
|
D | SparseMatrixMatMul.pbtxt | 41 name: "adjoint_b"
|
/external/tensorflow/tensorflow/core/ops/ |
D | sparse_csr_matrix_ops.cc | 255 bool adjoint_b = false; in __anon59a9d6c30702() local 257 TF_RETURN_IF_ERROR(c->GetAttr("adjoint_b", &adjoint_b)); in __anon59a9d6c30702() 262 if (adjoint_b && transpose_b) { in __anon59a9d6c30702() 267 transpose_b = transpose_b || adjoint_b; in __anon59a9d6c30702() 413 bool adjoint_b = false; in __anon59a9d6c30a02() local 415 TF_RETURN_IF_ERROR(c->GetAttr("adjoint_b", &adjoint_b)); in __anon59a9d6c30a02() 419 } else if (adjoint_b && transpose_b) { in __anon59a9d6c30a02() 424 transpose_b = transpose_b || adjoint_b; in __anon59a9d6c30a02()
|
D | sparse_csr_matrix_ops_test.cc | 109 bool adjoint_b, bool transpose_output) { in TEST() argument 116 .Attr("adjoint_b", adjoint_b) in TEST() 233 bool adjoint_b) { in TEST() argument 240 .Attr("adjoint_b", adjoint_b) in TEST()
|
/external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
D | linear_operator_low_rank_update_test.py | 137 u, math_ops.matmul(diag_update_mat, v, adjoint_b=True)) 141 matrix = base_diag_mat + math_ops.matmul(u, v, adjoint_b=True) 146 u, math_ops.matmul(diag_update_mat, u, adjoint_b=True)) 150 matrix = base_diag_mat + math_ops.matmul(u, u, adjoint_b=True)
|
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)
|
/external/tensorflow/tensorflow/python/ops/linalg/ |
D | linear_operator_util.py | 391 adjoint_b=False): argument 430 b_eq_sz_ = b.shape[-2 if adjoint_b or transpose_b else -1] 446 if adjoint_b:
|