| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| H A D | batch_matmul_op_test.py | 45 def _npBatchMatmul(self, x, y, adjoint_a, adjoint_b): argument
54 def _compare(self, x_in, y_in, adjoint_a, adjoint_b, static_shape): argument
74 def _testNonEmpty(self, dtype, adjoint_a, adjoint_b, use_static_shape): argument
94 def _testBroadcasting(self, dtype, adjoint_a, adjoint_b, use_static_shape): argument
112 def _testEmpty(self, dtype, adjoint_a, adjoint_b, use_static_shape): argument
127 def _GetBatchMatmulOpTest(dtype, adjoint_a, adjoint_b, use_static_shape): argument
138 def _GetBatchMatmulOpBroadcastingTest(dtype, adjoint_a, adjoint_b, argument
153 def _checkGrad(self, x_in, y_in, adjoint_a, adjoint_b): argument
174 def _compare(self, a_shape, b_shape, dtype, adjoint_a, adjoint_b): argument
181 def _GetBatchMatmulGradientTest(dtype, adjoint_a, adjoint_b): argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| H A D | sparse_tensor_dense_matmul_op_test.py | 50 adjoint_b=False, argument
258 adjoint_b): argument
288 adjoint_b): argument
318 adjoint_b, argument
|
| H A D | sparse_tensor_dense_matmul_grad_test.py | 60 def _testGradients(self, adjoint_a, adjoint_b, name, values_dtype, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| H A D | csr_sparse_matrix_test.py | 147 def _testSparseSparse(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
187 def _testSparseDense(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
225 def _testDenseSparse(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
|
| H A D | csr_sparse_matrix_sparse_mat_mul_grad_test.py | 60 adjoint_b): argument
|
| H A D | csr_sparse_matrix_dense_mat_mul_grad_test.py | 56 adjoint_b, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_tensor_dense_matmul_op_test.cc | 28 bool adjoint_b) { in SparseTensorDenseMatMulNode()
43 bool adjoint_a, bool adjoint_b) { in SparseTensorDenseMatmul()
|
| H A D | matmul_op_test.cc | 496 bool adjoint_b, DataType type) { in BatchMatmul()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
| H A D | sparse_csr_matrix_ops.py | 116 adjoint_b): argument
146 adjoint_b=False, argument
|
| H A D | sparse_csr_matrix_grad.py | 138 adjoint_b=False): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | sparse_csr_matrix_ops_test.cc | 109 bool adjoint_b, bool transpose_output) { in TEST()
233 bool adjoint_b) { in TEST()
|
| H A D | sparse_csr_matrix_ops.cc | 255 bool adjoint_b = false; in __anona88575120702() local
413 bool adjoint_b = false; in __anona88575120a02() local
|
| H A D | sparse_ops_test.cc | 186 auto set_adjoints = [&op](bool adjoint_a, bool adjoint_b) { in TEST()
|
| H A D | sparse_ops.cc | 109 bool adjoint_b; in __anone91b66370402() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/ |
| H A D | linear_operator_util.py | 388 adjoint_b=False): argument
|
| H A D | linear_operator.py | 1480 adjoint_b=False, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/ |
| H A D | sparse_mat_mul_op.cc | 303 bool adjoint_b; in CSRSparseMatMulGPUOp() local
|
| H A D | mat_mul_op.cc | 90 bool adjoint_b; in CSRMatMulOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_math_ops.py | 796 adjoint_b=False, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | sparse_ops.py | 2419 adjoint_b=False, argument
|
| H A D | math_ops.py | 3496 adjoint_b=False, argument
|