| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | batch_matmul_op_test.py | 33 def _npBatchMatmul(self, x, y, adjoint_a, adjoint_b): argument
35 d0 = x.shape[-2] if not adjoint_a else x.shape[-1]
45 if adjoint_a:
79 def _compare(self, x_in, y_in, adjoint_a, adjoint_b, static_shape=True): argument
82 x = x_in if not adjoint_a else x_in.reshape(x_t_shape)
88 z0 = math_ops.matmul(x, y, adjoint_a=adjoint_a, adjoint_b=adjoint_b)
94 x_ph, y_ph, adjoint_a=adjoint_a, adjoint_b=adjoint_b)
96 z1 = self._npBatchMatmul(x, y, adjoint_a, adjoint_b)
106 def _testNonEmpty(self, dtype, adjoint_a, adjoint_b, use_static_shape): argument
111 self._rand(b_shape, dtype), adjoint_a, adjoint_b, use_static_shape)
[all …]
|
| D | sparse_tensor_dense_matmul_op_test.py | 52 adjoint_a=False, argument
56 if adjoint_a:
72 sp_x_value, y, adjoint_a=adjoint_a, adjoint_b=adjoint_b)
76 adjoint_a=adjoint_a,
158 sparse_t, dense_t, adjoint_a=True).eval()
163 sparse_t, dense_t, adjoint_a=True).eval()
197 sparse_t, dense_t, adjoint_a=True).eval())
203 sparse_t, dense_t, adjoint_a=True).eval())
219 self._testMatmul(x, y, adjoint_a=False, adjoint_b=False)
220 self._testMatmul(x.transpose(), y, adjoint_a=True, adjoint_b=False)
[all …]
|
| D | sparse_tensor_dense_matmul_grad_test.py | 61 def _testGradients(self, adjoint_a, adjoint_b, name, values_dtype, argument
67 adjoint=adjoint_a,
73 sp_t, dense_t, adjoint_a=adjoint_a, adjoint_b=adjoint_b, name=name)
85 for adjoint_a in [True, False]:
88 adjoint_a, adjoint_b, values_dtype.__name__, indices_dtype.__name__)
89 self._testGradients(adjoint_a, adjoint_b, name, values_dtype,
|
| D | matmul_op_test.py | 236 for adjoint_a, transpose_a in trans_options:
239 use_static_shape, dtype.__name__, m, n, k, adjoint_a,
246 adjoint_a=adjoint_a,
255 adjoint_a=adjoint_a,
|
| D | cholesky_op_test.py | 49 middle = math_ops.matmul(l, grad, adjoint_a=True)
54 math_ops.matmul(l_inverse, middle, adjoint_a=True), l_inverse)
63 middle = math_ops.matmul(l, grad, adjoint_a=True)
181 matrix1 = math_ops.matmul(matrix1, matrix1, adjoint_a=True)
182 matrix2 = math_ops.matmul(matrix2, matrix2, adjoint_a=True)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_grad.py | 43 ainv, math_ops.matmul(grad, ainv, adjoint_b=True), adjoint_a=True)
72 middle = math_ops.matmul(l, grad, adjoint_a=True)
78 math_ops.matmul(l_inverse, middle, adjoint_a=True), l_inverse)
97 qdq = math_ops.matmul(q, dq, adjoint_a=True)
118 adjoint_a = op.get_attr("adjoint")
120 grad_b = linalg_ops.matrix_solve(a, grad, adjoint=not adjoint_a)
121 if adjoint_a:
182 a1 = math_ops.matmul(tmp, a, adjoint_a=True)
184 a2 = grad - math_ops.matmul(a, grad_b, adjoint_a=True)
211 adjoint_a = op.get_attr("adjoint")
[all …]
|
| D | linalg_ops.py | 65 matrix, matrix, adjoint_a=first_kind, adjoint_b=not first_kind)
290 return cholesky_solve(chol, math_ops.matmul(matrix, rhs, adjoint_a=True))
296 return math_ops.matmul(matrix, cholesky_solve(chol, rhs), adjoint_a=True)
|
| D | math_grad.py | 1159 grad_x = math_ops.matmul(grad, y, adjoint_a=False, adjoint_b=True)
1160 grad_y = math_ops.matmul(x, grad, adjoint_a=True, adjoint_b=False)
1162 grad_x = math_ops.matmul(grad, y, adjoint_a=False, adjoint_b=False)
1163 grad_y = math_ops.matmul(grad, x, adjoint_a=True, adjoint_b=False)
1166 grad_x = math_ops.matmul(y, grad, adjoint_a=False, adjoint_b=True)
1167 grad_y = math_ops.matmul(x, grad, adjoint_a=False, adjoint_b=False)
1169 grad_x = math_ops.matmul(y, grad, adjoint_a=True, adjoint_b=True)
1170 grad_y = math_ops.matmul(grad, x, adjoint_a=True, adjoint_b=True)
|
| D | sparse_ops.py | 1626 adjoint_a=False, argument
1836 adjoint_a=adjoint_a,
|
| D | math_ops.py | 1902 adjoint_a=False, argument
2008 if transpose_a and adjoint_a:
2025 adjoint_a = True
2030 a, b, adj_x=adjoint_a, adj_y=adjoint_b, name=name)
2035 if adjoint_a:
|
| D | sparse_grad.py | 155 a_indices, a_values, a_shape, grad, adjoint_a=not adj_a)
|
| /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 | 24 static Graph* BatchMatmul(int b, int m, int k, int n, bool adjoint_a, in BatchMatmul() argument
27 Tensor in0(type, adjoint_a ? TensorShape({b, k, m}) : TensorShape({b, m, k})); in BatchMatmul()
32 test::graph::Constant(g, in1), adjoint_a, adjoint_b); in BatchMatmul()
|
| /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:
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_util.py | 202 adjoint_a=False, argument
286 adjoint_a=adjoint_a,
|
| D | linear_operator_low_rank_update.py | 368 uh_x = math_ops.matmul(u, x, adjoint_a=True, adjoint_b=adjoint_arg)
373 vh_x = math_ops.matmul(v, x, adjoint_a=True, adjoint_b=adjoint_arg)
434 vh_linv_rhs = math_ops.matmul(v, linv_rhs, adjoint_a=True)
457 vh_linv_u = math_ops.matmul(self.v, linv_u, adjoint_a=True)
|
| D | linear_operator_full_matrix.py | 181 self._matrix, x, adjoint_a=adjoint, adjoint_b=adjoint_arg)
|
| D | linear_operator_lower_triangular.py | 199 self._tril, x, adjoint_a=adjoint, adjoint_b=adjoint_arg)
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/ |
| D | kalman_filter.py | 256 adjoint_a=True),
259 kalman_gain_transposed, observation_model, adjoint_a=True)
287 kalman_gain_transposed, observation_noise, adjoint_a=True),
|
| /external/tensorflow/tensorflow/contrib/solvers/python/ops/ |
| D | util.py | 44 apply=lambda v: math_ops.matmul(matrix, v, adjoint_a=False),
45 apply_adjoint=lambda v: math_ops.matmul(matrix, v, adjoint_a=True))
|
| D | lanczos.py | 124 v -= math_ops.matmul(basis_vec, v, adjoint_a=True) * basis_vec
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops_test.cc | 167 auto set_adjoints = [&op](bool adjoint_a, bool adjoint_b) { in TEST() argument
173 .Attr("adjoint_a", adjoint_a) in TEST()
|
| D | sparse_ops.cc | 107 bool adjoint_a; in __anone6e195410402() local
109 TF_RETURN_IF_ERROR(c->GetAttr("adjoint_a", &adjoint_a)); in __anone6e195410402()
113 DimensionHandle output_left = c->Dim(a_shape, adjoint_a ? 1 : 0); in __anone6e195410402()
114 DimensionHandle inner_left = c->Dim(a_shape, adjoint_a ? 0 : 1); in __anone6e195410402()
|
| /external/tensorflow/tensorflow/contrib/solvers/python/kernel_tests/ |
| D | lanczos_test.py | 64 atu = math_ops.matmul(a, lbd.u[:, :-1], adjoint_a=True)
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | matrix_triangular_solve_op_test.py | 56 verification = math_ops.matmul(placeholder_ca, x, adjoint_a=adjoint)
|