| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | batch_matmul_op_test.py | 49 def _npBatchMatmul(self, x, y, adjoint_a, adjoint_b): argument
58 def _compare(self, x_in, y_in, adjoint_a, adjoint_b, static_shape): argument
78 def _testNonEmpty(self, dtype, adjoint_a, adjoint_b, use_static_shape): argument
98 def _testBroadcasting(self, dtype, adjoint_a, adjoint_b, use_static_shape): argument
116 def _testEmpty(self, dtype, adjoint_a, adjoint_b, use_static_shape): argument
131 def _GetBatchMatmulOpTest(dtype, adjoint_a, adjoint_b, use_static_shape): argument
142 def _GetBatchMatmulOpBroadcastingTest(dtype, adjoint_a, adjoint_b, argument
157 def _checkGrad(self, x_in, y_in, adjoint_a, adjoint_b): argument
178 def _compare(self, a_shape, b_shape, dtype, adjoint_a, adjoint_b): argument
185 def _GetBatchMatmulGradientTest(dtype, adjoint_a, adjoint_b): argument
[all …]
|
| D | sparse_tensor_dense_matmul_op_test.py | 53 adjoint_a=False, argument
245 def _sparse_tensor_dense_vs_dense_matmul_benchmark_dense(x, y, adjoint_a, argument
275 y, adjoint_a, argument
305 adjoint_a, argument
|
| D | sparse_tensor_dense_matmul_grad_test.py | 62 def _testGradients(self, adjoint_a, adjoint_b, name, values_dtype, argument
|
| /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
191 def _testSparseDense(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
229 def _testDenseSparse(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
|
| D | csr_sparse_matrix_sparse_mat_mul_grad_test.py | 62 transpose_b, adjoint_a, argument
|
| D | csr_sparse_matrix_dense_mat_mul_grad_test.py | 62 transpose_b, adjoint_a, adjoint_b, argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_tensor_dense_matmul_op_test.cc | 27 Node* a_shape, Node* b, bool adjoint_a, in SparseTensorDenseMatMulNode()
43 bool adjoint_a, bool adjoint_b) { in SparseTensorDenseMatmul()
|
| D | matmul_op_test.cc | 488 static Graph* BatchMatmul(int b, int m, int k, int n, bool adjoint_a, in BatchMatmul()
|
| /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
151 adjoint_a=False, argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_csr_matrix_ops_test.cc | 109 bool adjoint_b, bool transpose_output) { in TEST()
233 bool adjoint_b) { in TEST()
|
| D | sparse_csr_matrix_ops.cc | 254 bool adjoint_a = false; in __anon59a9d6c30702() local
412 bool adjoint_a = false; in __anon59a9d6c30a02() local
|
| D | sparse_ops_test.cc | 186 auto set_adjoints = [&op](bool adjoint_a, bool adjoint_b) { in TEST()
|
| D | sparse_ops.cc | 106 bool adjoint_a; in __anon36a8fc880402() local
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_util.py | 390 adjoint_a=False, argument
|
| D | linear_operator.py | 1216 adjoint_a=False, argument
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_mat_mul_op.cc | 298 bool adjoint_a; in CSRSparseMatMulGPUOp() local
|
| D | mat_mul_op.cc | 84 bool adjoint_a; in CSRMatMulOp() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 2425 adjoint_a=False, argument
|
| D | math_ops.py | 3266 adjoint_a=False, argument
3460 adjoint_a=False, argument
|