/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, 163 sparse_t, dense_t, adjoint_a=True)) 169 sparse_t, dense_t, adjoint_a=True)) 203 sparse_t, dense_t, adjoint_a=True)) 209 sparse_t, dense_t, adjoint_a=True)) 225 self._testMatmul(x, y, adjoint_a=False, adjoint_b=False) 226 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 | 253 for adjoint_a, transpose_a in trans_options: 256 use_static_shape, dtype.__name__, m, n, k, adjoint_a, 263 adjoint_a=adjoint_a, 272 adjoint_a=adjoint_a,
|
D | cholesky_op_test.py | 45 middle = math_ops.matmul(l, grad, adjoint_a=True) 50 math_ops.matmul(l_inverse, middle, adjoint_a=True), l_inverse) 59 middle = math_ops.matmul(l, grad, adjoint_a=True) 183 matrix1 = math_ops.matmul(matrix1, matrix1, adjoint_a=True) 184 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] 173 adjoint_a=adjoint_a, 180 adjoint_a=adjoint_a, 194 def _testSparseDense(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument 199 dense_shape_a = [5, 13, 7] if transpose_a or adjoint_a else [5, 7, 13] 216 adjoint_a=adjoint_a, 223 adjoint_a=adjoint_a, 235 def _testDenseSparse(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument 240 dense_shape_a = [5, 13, 7] if transpose_a or adjoint_a else [5, 7, 13] [all …]
|
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_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_ops_test.py | 560 for (adjoint_a, adjoint_b) in ((False, False), (False, True), 562 if (transpose_a and adjoint_a) or (transpose_b and adjoint_b): 569 if transpose_a or adjoint_a: 581 (transpose_a, transpose_b, adjoint_a, adjoint_b)) 590 adjoint_a=adjoint_a, 597 adjoint_a=adjoint_a, 622 for (adjoint_a, adjoint_b) in ((False, False), (False, True), 624 if (transpose_a and adjoint_a) or (transpose_b and adjoint_b): 631 if transpose_a or adjoint_a: 643 (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()
|
D | batch_matmul_op_test.cc | 53 static Graph* BatchMatmul(int b, int m, int k, int n, bool adjoint_a, in BatchMatmul() argument 56 Tensor in0(type, adjoint_a ? TensorShape({b, k, m}) : TensorShape({b, m, k})); in BatchMatmul() 61 test::graph::Constant(g, in1), adjoint_a, adjoint_b); in BatchMatmul()
|
/external/tensorflow/tensorflow/python/ops/ |
D | linalg_grad.py | 55 ainv, math_ops.matmul(grad, ainv, adjoint_b=True), adjoint_a=True) 471 middle = math_ops.matmul(l, grad, adjoint_a=True) 477 math_ops.matmul(l_inverse, middle, adjoint_a=True), l_inverse) 496 qdq = math_ops.matmul(q, dq, adjoint_a=True) 517 adjoint_a = op.get_attr("adjoint") 519 grad_b = linalg_ops.matrix_solve(a, grad, adjoint=not adjoint_a) 520 if adjoint_a: 581 a1 = math_ops.matmul(tmp, a, adjoint_a=True) 583 a2 = grad - math_ops.matmul(a, grad_b, adjoint_a=True) 611 adjoint_a = op.get_attr("adjoint") [all …]
|
D | sparse_ops_test.py | 160 independent_dense_tf, sp, adjoint_a=False, adjoint_b=False) 165 independent_dense_tf, sp, adjoint_a=False, adjoint_b=True) 171 independent_dense_tf, sp, adjoint_a=True, adjoint_b=False) 177 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/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() 524 bool adjoint_a, bool transpose_b) in CSRSparseSparseMatrixMatMul() 529 adjoint_a_(adjoint_a), in CSRSparseSparseMatrixMatMul() 533 transA_ = transpose_a ? (adjoint_a ? CUSPARSE_OPERATION_TRANSPOSE in CSRSparseSparseMatrixMatMul() 540 ? (adjoint_a ? HIPSPARSE_OPERATION_TRANSPOSE in CSRSparseSparseMatrixMatMul()
|
/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 __anond96a6a670702() local 256 TF_RETURN_IF_ERROR(c->GetAttr("adjoint_a", &adjoint_a)); in __anond96a6a670702() 258 if (adjoint_a && transpose_a) { in __anond96a6a670702() 266 transpose_a = transpose_a || adjoint_a; in __anond96a6a670702() 412 bool adjoint_a = false; in __anond96a6a670a02() local 414 TF_RETURN_IF_ERROR(c->GetAttr("adjoint_a", &adjoint_a)); in __anond96a6a670a02() 416 if (adjoint_a && transpose_a) { in __anond96a6a670a02() 423 transpose_a = transpose_a || adjoint_a; in __anond96a6a670a02()
|
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 | 352 uh_x = math_ops.matmul(u, x, adjoint_a=True, adjoint_b=adjoint_arg) 357 vh_x = math_ops.matmul(v, x, adjoint_a=True, adjoint_b=adjoint_arg) 421 vh_linv_rhs = math_ops.matmul(v, linv_rhs, adjoint_a=True) 444 vh_linv_u = math_ops.matmul(self.v, linv_u, adjoint_a=True)
|
D | linear_operator_util.py | 365 matrix, rhs, adjoint_a=adjoint) 380 adjoint_a=False, argument 419 a_domain_sz_ = a.shape[-2 if adjoint_a or transpose_a else -1] 432 if adjoint_a:
|
/external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
D | linear_operator_test.py | 91 self._matrix, x, adjoint_a=adjoint, adjoint_b=adjoint_arg) 348 operator, rhs, adjoint_a=adjoint, adjoint_b=adjoint_arg) 370 lhs, mat, adjoint_a=adjoint, adjoint_b=adjoint_arg) 372 lhs, right_operator, adjoint_a=adjoint, adjoint_b=adjoint_arg)
|