Home
last modified time | relevance | path

Searched refs:adjoint_a (Results 1 – 25 of 49) sorted by relevance

12

/external/tensorflow/tensorflow/python/kernel_tests/
Dbatch_matmul_op_test.py34 def _npBatchMatmul(self, x, y, adjoint_a, adjoint_b): argument
36 d0 = x.shape[-2] if not adjoint_a else x.shape[-1]
46 if adjoint_a:
80 def _compare(self, x_in, y_in, adjoint_a, adjoint_b, static_shape=True): argument
83 x = x_in if not adjoint_a else x_in.reshape(x_t_shape)
89 z0 = math_ops.matmul(x, y, adjoint_a=adjoint_a, adjoint_b=adjoint_b)
95 x_ph, y_ph, adjoint_a=adjoint_a, adjoint_b=adjoint_b)
97 z1 = self._npBatchMatmul(x, y, adjoint_a, adjoint_b)
107 def _testNonEmpty(self, dtype, adjoint_a, adjoint_b, use_static_shape): argument
112 self._rand(b_shape, dtype), adjoint_a, adjoint_b, use_static_shape)
[all …]
Dsparse_tensor_dense_matmul_op_test.py53 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 …]
Dsparse_tensor_dense_matmul_grad_test.py62 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,
Dmatmul_op_test.py249 for adjoint_a, transpose_a in trans_options:
252 use_static_shape, dtype.__name__, m, n, k, adjoint_a,
259 adjoint_a=adjoint_a,
268 adjoint_a=adjoint_a,
Dcholesky_op_test.py51 middle = math_ops.matmul(l, grad, adjoint_a=True)
56 math_ops.matmul(l_inverse, middle, adjoint_a=True), l_inverse)
65 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/ops/
Dlinalg_grad.py43 ainv, math_ops.matmul(grad, ainv, adjoint_b=True), adjoint_a=True)
148 middle = math_ops.matmul(l, grad, adjoint_a=True)
154 math_ops.matmul(l_inverse, middle, adjoint_a=True), l_inverse)
173 qdq = math_ops.matmul(q, dq, adjoint_a=True)
194 adjoint_a = op.get_attr("adjoint")
196 grad_b = linalg_ops.matrix_solve(a, grad, adjoint=not adjoint_a)
197 if adjoint_a:
258 a1 = math_ops.matmul(tmp, a, adjoint_a=True)
260 a2 = grad - math_ops.matmul(a, grad_b, adjoint_a=True)
287 adjoint_a = op.get_attr("adjoint")
[all …]
Dlinalg_ops.py66 matrix, matrix, adjoint_a=first_kind, adjoint_b=not first_kind)
269 return cholesky_solve(chol, math_ops.matmul(matrix, rhs, adjoint_a=True))
275 return math_ops.matmul(matrix, cholesky_solve(chol, rhs), adjoint_a=True)
Dmath_grad.py1254 grad_x = math_ops.matmul(grad, y, adjoint_a=False, adjoint_b=True)
1255 grad_y = math_ops.matmul(x, grad, adjoint_a=True, adjoint_b=False)
1257 grad_x = math_ops.matmul(grad, y, adjoint_a=False, adjoint_b=False)
1258 grad_y = math_ops.matmul(grad, x, adjoint_a=True, adjoint_b=False)
1261 grad_x = math_ops.matmul(y, grad, adjoint_a=False, adjoint_b=True)
1262 grad_y = math_ops.matmul(x, grad, adjoint_a=False, adjoint_b=False)
1264 grad_x = math_ops.matmul(y, grad, adjoint_a=True, adjoint_b=True)
1265 grad_y = math_ops.matmul(grad, x, adjoint_a=True, adjoint_b=True)
Dmath_ops.py2407 adjoint_a=False, argument
2513 if transpose_a and adjoint_a:
2537 adjoint_a = True
2542 a, b, adj_x=adjoint_a, adj_y=adjoint_b, name=name)
2547 if adjoint_a:
2587 adjoint_a=False, argument
2675 adjoint_a=adjoint_a,
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_util.py231 adjoint_a=False, argument
323 adjoint_a=adjoint_a,
334 adjoint_a=adjoint_a and still_need_to_transpose,
350 matrix, rhs, adjoint_a=adjoint)
393 matrix, rhs, adjoint_a=adjoint)
416 adjoint_a=False, argument
455 a_domain_sz_ = a.shape[-2 if adjoint_a or transpose_a else -1]
468 if adjoint_a:
Dlinear_operator_low_rank_update.py358 u, x, adjoint_a=True, adjoint_b=adjoint_arg)
365 v, x, adjoint_a=True, adjoint_b=adjoint_arg)
429 v, linv_rhs, adjoint_a=True)
454 self.v, linv_u, adjoint_a=True)
Dlinear_operator_full_matrix.py180 self._matrix, x, adjoint_a=adjoint, adjoint_b=adjoint_arg)
Dlinear_operator_lower_triangular.py195 self._tril, x, adjoint_a=adjoint, adjoint_b=adjoint_arg)
/external/tensorflow/tensorflow/core/kernels/
Dsparse_tensor_dense_matmul_op_test.cc27 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()
Dbatch_matmul_op_test.cc24 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/
Dapi_def_SparseTensorDenseMatMul.pbtxt28 name: "adjoint_a"
46 if adjoint_a == false:
49 if adjoint_a == true:
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/
Dkalman_filter.py255 adjoint_a=True),
258 kalman_gain_transposed, observation_model, adjoint_a=True)
286 kalman_gain_transposed, observation_noise, adjoint_a=True),
/external/tensorflow/tensorflow/contrib/solvers/python/ops/
Dutil.py44 apply=lambda v: math_ops.matmul(matrix, v, adjoint_a=False),
45 apply_adjoint=lambda v: math_ops.matmul(matrix, v, adjoint_a=True))
/external/tensorflow/tensorflow/core/ops/
Dsparse_ops_test.cc186 auto set_adjoints = [&op](bool adjoint_a, bool adjoint_b) { in TEST() argument
192 .Attr("adjoint_a", adjoint_a) in TEST()
Dsparse_ops.cc106 bool adjoint_a; in __anona7ffb82a0402() local
108 TF_RETURN_IF_ERROR(c->GetAttr("adjoint_a", &adjoint_a)); in __anona7ffb82a0402()
112 DimensionHandle output_left = c->Dim(a_shape, adjoint_a ? 1 : 0); in __anona7ffb82a0402()
113 DimensionHandle inner_left = c->Dim(a_shape, adjoint_a ? 0 : 1); in __anona7ffb82a0402()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.sparse.pbtxt45 …argspec: "args=[\'sp_a\', \'b\', \'adjoint_a\', \'adjoint_b\', \'name\'], varargs=None, keywords=N…
117 …argspec: "args=[\'sp_a\', \'b\', \'adjoint_a\', \'adjoint_b\', \'name\'], varargs=None, keywords=N…
Dtensorflow.linalg.pbtxt149 …argspec: "args=[\'a\', \'b\', \'transpose_a\', \'transpose_b\', \'adjoint_a\', \'adjoint_b\', \'a_…
153 …argspec: "args=[\'a\', \'b\', \'transpose_a\', \'adjoint_a\', \'a_is_sparse\', \'b_is_sparse\', \'…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.linalg.pbtxt149 …argspec: "args=[\'a\', \'b\', \'transpose_a\', \'transpose_b\', \'adjoint_a\', \'adjoint_b\', \'a_…
153 …argspec: "args=[\'a\', \'b\', \'transpose_a\', \'adjoint_a\', \'a_is_sparse\', \'b_is_sparse\', \'…
/external/tensorflow/tensorflow/contrib/solvers/python/kernel_tests/
Dlanczos_test.py65 atu = math_ops.matmul(a, lbd.u[:, :-1], adjoint_a=True)
/external/tensorflow/tensorflow/compiler/tests/
Dqr_op_test.py64 xx = math_ops.matmul(x, x, adjoint_a=True)

12