Home
last modified time | relevance | path

Searched refs:adjoint_b (Results 1 – 25 of 63) sorted by relevance

123

/external/tensorflow/tensorflow/python/kernel_tests/
Dbatch_matmul_op_test.py49 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 …]
Dsparse_tensor_dense_matmul_op_test.py54 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)
225 self._testMatmul(x, y, adjoint_a=False, adjoint_b=False)
226 self._testMatmul(x.transpose(), y, adjoint_a=True, adjoint_b=False)
227 self._testMatmul(x, y.transpose(), adjoint_a=False, adjoint_b=True)
229 x.transpose(), y.transpose(), adjoint_a=True, adjoint_b=True)
243 for adjoint_b in [True, False]:
250 y = y.transpose() if adjoint_b else y
[all …]
Dsparse_tensor_dense_matmul_grad_test.py62 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,
Dmatmul_op_test.py254 for adjoint_b, transpose_b in trans_options:
257 transpose_a, adjoint_b, transpose_b)
265 adjoint_b=adjoint_b,
274 adjoint_b=adjoint_b,
/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/
Dcsr_sparse_matrix_test.py151 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]
174 adjoint_b=adjoint_b)
181 adjoint_b=adjoint_b)
194 def _testSparseDense(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
200 dense_shape_b = [5, 15, 13] if transpose_b or adjoint_b else [5, 13, 15]
217 adjoint_b=adjoint_b)
224 adjoint_b=adjoint_b)
235 def _testDenseSparse(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
241 dense_shape_b = [5, 15, 13] if transpose_b or adjoint_b else [5, 13, 15]
[all …]
Dcsr_sparse_matrix_sparse_mat_mul_grad_test.py63 adjoint_b): argument
78 if transpose_b or adjoint_b:
80 if adjoint_b:
93 adjoint_b=adjoint_b,
Dcsr_sparse_matrix_dense_mat_mul_grad_test.py62 transpose_b, adjoint_a, adjoint_b, argument
77 if transpose_b or adjoint_b:
79 if adjoint_b:
91 adjoint_b=adjoint_b,
Dcsr_sparse_matrix_ops_test.py560 for (adjoint_a, adjoint_b) in ((False, False), (False, True),
562 if (transpose_a and adjoint_a) or (transpose_b and adjoint_b):
571 if transpose_b or adjoint_b:
581 (transpose_a, transpose_b, adjoint_a, adjoint_b))
591 adjoint_b=adjoint_b)
598 adjoint_b=adjoint_b)
622 for (adjoint_a, adjoint_b) in ((False, False), (False, True),
624 if (transpose_a and adjoint_a) or (transpose_b and adjoint_b):
633 if transpose_b or adjoint_b:
643 (transpose_a, transpose_b, adjoint_a, adjoint_b))
[all …]
/external/tensorflow/tensorflow/python/ops/linalg/sparse/
Dsparse_csr_matrix_ops.py122 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 …]
Dsparse_csr_matrix_grad.py121 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/
Dsparse_tensor_dense_matmul_op_test.cc28 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()
Dbatch_matmul_op_test.cc54 bool adjoint_b, DataType type) { in BatchMatmul() argument
58 Tensor in1(type, adjoint_b ? TensorShape({b, n, k}) : TensorShape({b, k, n})); in BatchMatmul()
61 test::graph::Constant(g, in1), adjoint_a, adjoint_b); in BatchMatmul()
/external/tensorflow/tensorflow/python/ops/
Dlinalg_grad.py55 ainv, math_ops.matmul(grad, ainv, adjoint_b=True), adjoint_a=True)
498 rdr = math_ops.matmul(r, dr, adjoint_b=True)
521 grad_a = -math_ops.matmul(c, grad_b, adjoint_b=True)
523 grad_a = -math_ops.matmul(grad_b, c, adjoint_b=True)
556 xzt = math_ops.matmul(x, z, adjoint_b=True)
558 grad_a = -math_ops.matmul(a, zx_sym) + math_ops.matmul(b, z, adjoint_b=True)
584 a2 = math_ops.matmul(tmp, a2, adjoint_b=True)
617 grad_a = -math_ops.matmul(c, grad_b, adjoint_b=True)
619 grad_a = -math_ops.matmul(grad_b, c, adjoint_b=True)
662 adjoint_b=True))
[all …]
Dsparse_ops_test.py160 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)
Dmath_grad.py1785 grad_x = math_ops.matmul(grad, y, adjoint_a=False, adjoint_b=True)
1786 grad_y = math_ops.matmul(x, grad, adjoint_a=True, adjoint_b=False)
1788 grad_x = math_ops.matmul(grad, y, adjoint_a=False, adjoint_b=False)
1789 grad_y = math_ops.matmul(grad, x, adjoint_a=True, adjoint_b=False)
1792 grad_x = math_ops.matmul(y, grad, adjoint_a=False, adjoint_b=True)
1793 grad_y = math_ops.matmul(x, grad, adjoint_a=False, adjoint_b=False)
1795 grad_x = math_ops.matmul(y, grad, adjoint_a=True, adjoint_b=True)
1796 grad_y = math_ops.matmul(grad, x, adjoint_a=True, adjoint_b=True)
1811 grad_x = math_ops.matmul(grad, y, adjoint_a=False, adjoint_b=True)
1812 grad_y = math_ops.matmul(x, grad, adjoint_a=True, adjoint_b=False)
[all …]
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SparseMatrixSparseMatMul.pbtxt29 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_v1/
DSparseTensorDenseMatMul.pbtxt35 name: "adjoint_b"
89 name: "adjoint_b"
DSparseMatrixSparseMatMul.pbtxt49 name: "adjoint_b"
DSparseMatrixMatMul.pbtxt41 name: "adjoint_b"
/external/tensorflow/tensorflow/core/ops/
Dsparse_csr_matrix_ops.cc255 bool adjoint_b = false; in __anond96a6a670702() local
257 TF_RETURN_IF_ERROR(c->GetAttr("adjoint_b", &adjoint_b)); in __anond96a6a670702()
262 if (adjoint_b && transpose_b) { in __anond96a6a670702()
267 transpose_b = transpose_b || adjoint_b; in __anond96a6a670702()
413 bool adjoint_b = false; in __anond96a6a670a02() local
415 TF_RETURN_IF_ERROR(c->GetAttr("adjoint_b", &adjoint_b)); in __anond96a6a670a02()
419 } else if (adjoint_b && transpose_b) { in __anond96a6a670a02()
424 transpose_b = transpose_b || adjoint_b; in __anond96a6a670a02()
Dsparse_csr_matrix_ops_test.cc109 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/
Dlinear_operator_low_rank_update_test.py137 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)
Dlinear_operator_test.py91 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)
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_util.py381 adjoint_b=False): argument
420 b_eq_sz_ = b.shape[-2 if adjoint_b or transpose_b else -1]
436 if adjoint_b:
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_mat_mul_op.cc303 bool adjoint_b; in CSRSparseMatMulGPUOp() local
304 OP_REQUIRES_OK(c, c->GetAttr("adjoint_b", &adjoint_b)); in CSRSparseMatMulGPUOp()
305 OP_REQUIRES(c, !(adjoint_b && transpose_b_), in CSRSparseMatMulGPUOp()
309 conjugate_b_ = adjoint_b; in CSRSparseMatMulGPUOp()
311 transpose_b_ = transpose_b_ || adjoint_b; in CSRSparseMatMulGPUOp()

123