| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantized_matmul_op_test.cc | 95 const bool transpose_b = false; in TEST_F() local
145 const bool transpose_b = false; in TEST_F() local
184 const bool transpose_b = false; in TEST_F() local
223 const bool transpose_b = false; in TEST_F() local
312 const bool transpose_b = false; in TEST_F() local
|
| H A D | matmul_op_test.cc | 102 bool transpose_b, Tensor* output, in RunMatMulWithBias()
121 bool transpose_a, bool transpose_b, const string& activation_type, in RunMatMulWithBiasAndActivation()
153 bool transpose_b, Tensor* output, in RunFusedMatMulOp()
222 bool transpose_b) { in VerifyMatMulWithBias()
244 bool transpose_b, in VerifyConv2DWithBiasAndActivation()
353 static Graph* Matmul(int m, int k, int n, bool transpose_a, bool transpose_b, in Matmul()
|
| H A D | sparse_matmul_op_test.cc | 52 bool transpose_b, bool a_sparse, bool b_sparse) { in SparseMatMulNode()
68 bool transpose_a, bool transpose_b) { in SparseMatMulHelper()
91 bool transpose_b) { in SparseMatMul()
|
| H A D | reference_gemm.h | 34 void ReferenceGemm(bool transpose_a, bool transpose_b, bool transpose_c, in ReferenceGemm()
|
| H A D | meta_support.cc | 221 bool transpose_b, const quint8* a_data, const quint8* b_data, in QuantizedGemm()
|
| H A D | quantized_conv_ops.cc | 387 const bool transpose_b = false; in operator ()() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | matmul_benchmark.py | 31 def build_graph(device, n, m, k, transpose_a, transpose_b, dtype): argument
67 def run_graph(self, device, n, m, k, transpose_a, transpose_b, num_iters, argument
114 def run_test_gpu(self, n, m, k, transpose_a, transpose_b, dtype, num_iters): 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 | 59 transpose_b, adjoint_a, argument
|
| H A D | csr_sparse_matrix_dense_mat_mul_grad_test.py | 54 transpose_b, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | benchmarks_test.py | 70 transpose_b=False, argument
402 def _benchmark_np_matmul(self, m, transpose_b, num_iters): argument
410 transpose_b, argument
416 def _benchmark_gen_math_ops_matmul(self, m, transpose_b, num_iters): argument
423 def _benchmark_tfe_py_fastpath_execute_matmul(self, m, transpose_b, argument
431 def _benchmark_tfe_py_execute_matmul(self, m, transpose_b, num_iters): argument
447 transpose_b, argument
490 def _benchmark_nested_defun_matmul(self, m, transpose_b, num_iters): argument
494 def outer(a, b, c, transpose_b): argument
505 transpose_b, argument
|
| /aosp_15_r20/external/gemmlowp/eight_bit_int_gemm/ |
| H A D | eight_bit_int_gemm.cc | 217 bool CanHandleMetaFastpath(bool transpose_a, bool transpose_b, bool transpose_c, in CanHandleMetaFastpath()
301 void EightBitIntGemm(bool transpose_a, bool transpose_b, bool transpose_c, in EightBitIntGemm()
338 void EightBitIntGemm(bool transpose_a, bool transpose_b, bool transpose_c, in EightBitIntGemm()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/sparse/ |
| H A D | sparse_csr_matrix_ops.py | 115 def matmul_shape_inference(a, b, c, transpose_a, transpose_b, adjoint_a, argument
144 transpose_b=False, argument
|
| H A D | sparse_csr_matrix_grad.py | 137 transpose_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 | 246 bool transpose_b = false; in __anona88575120702() local
409 bool transpose_b = false; in __anona88575120a02() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | mkl_testlib.cc | 27 bool transpose_b) { in oneDNNMatmul()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_matmul_op_benchmark.cc | 25 static Graph* Matmul(int m, int k, int n, bool transpose_a, bool transpose_b, in Matmul()
|
| H A D | mkl_matmul_op.cc | 93 bool transpose_b = dim_pair[0].second == 1; in Compute() local
|
| /aosp_15_r20/external/gemmlowp/test/ |
| H A D | test.cc | 35 void ReferenceEightBitIntGemm(bool transpose_a, bool transpose_b, in ReferenceEightBitIntGemm()
250 const bool transpose_b = RhsOrder == MapOrder::RowMajor; in Gemm() local
428 const bool transpose_b = kRhsOrder == MapOrder::RowMajor; in test_gemm_impl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/ |
| H A D | linear_operator_util.py | 386 transpose_b=False, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/ops/ |
| H A D | math_ops.cc | 110 bool transpose_a, bool transpose_b, const char* name, in MatMul()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/gradients/ |
| H A D | math_grad_test.cc | 248 bool transpose_b = transpose_b_vals[i]; in TEST_P() local
|
| /aosp_15_r20/frameworks/rs/cpu_ref/ |
| H A D | rsCpuIntrinsicBLAS.cpp | 885 bool transpose_b = false; in kernelBNNM() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes.cc | 4479 bool transpose_b) { in ConvertFullyConnectedImpl()
4588 bool transpose_b) { in ConvertMatMulImpl()
4662 bool transpose_b) { in ConvertMatMulHelper()
4681 bool transpose_a = false, transpose_b = false; in ConvertMatMul() local
4707 bool transpose_a = false, transpose_b = false; in ConvertBatchMatMul() local
|