| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantized_matmul_op_test.cc | 90 const bool transpose_a = true; in TEST_F() local
140 const bool transpose_a = true; in TEST_F() local
179 const bool transpose_a = true; in TEST_F() local
218 const bool transpose_a = true; in TEST_F() local
311 const bool transpose_a = true; in TEST_F() local
|
| H A D | matmul_op_test.cc | 101 const Tensor& bias_data, bool transpose_a, in RunMatMulWithBias()
121 bool transpose_a, bool transpose_b, const string& activation_type, in RunMatMulWithBiasAndActivation()
152 const std::vector<string>& fused_ops, bool transpose_a, in RunFusedMatMulOp()
221 void VerifyMatMulWithBias(int m, int k, int n, bool transpose_a, in VerifyMatMulWithBias()
243 void VerifyConv2DWithBiasAndActivation(int m, int k, int n, bool transpose_a, 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 | 51 Node* SparseMatMulNode(Graph* g, Node* in0, Node* in1, bool transpose_a, in SparseMatMulNode()
68 bool transpose_a, bool transpose_b) { in SparseMatMulHelper()
90 float sparsity_b, bool transpose_a, 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 | 220 void QuantizedGemm(OpKernelContext* tf_context, bool transpose_a, in QuantizedGemm()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | triangular_solve_expander.cc | 130 bool transpose_a, bool conjugate_a, in SolveWithInvertedDiagonalBlocks()
366 XlaOp a, XlaOp b, bool left_side, bool lower, bool transpose_a, in SolveByInvertingDiagonalBlocks()
410 XlaOp a, XlaOp b, bool left_side, bool lower, bool transpose_a, in SolveDirectly()
467 XlaOp a, XlaOp b, bool left_side, bool lower, bool transpose_a, in BuildTriangularSolve()
586 bool transpose_a = in ExpandInstruction() 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 | 58 def _testLargeBatchSparseMatrixSparseMatMulGrad(self, datatype, transpose_a, argument
|
| H A D | csr_sparse_matrix_dense_mat_mul_grad_test.py | 53 transpose_a, 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
143 transpose_a=False, argument
|
| H A D | sparse_csr_matrix_grad.py | 135 transpose_a=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 | 245 bool transpose_a = false; in __anona88575120702() local
408 bool transpose_a = false; in __anona88575120a02() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | mkl_testlib.cc | 26 Node* oneDNNMatmul(Graph* g, Node* in0, Node* in1, bool transpose_a, 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 | 92 bool transpose_a = dim_pair[0].first == 0; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | triangular_solve_thunk.cc | 90 se::blas::Transpose transpose_a, PrimitiveType type, in RunTriangulatSolve()
|
| /aosp_15_r20/external/gemmlowp/test/ |
| H A D | test.cc | 35 void ReferenceEightBitIntGemm(bool transpose_a, bool transpose_b, in ReferenceEightBitIntGemm()
249 const bool transpose_a = LhsOrder == MapOrder::RowMajor; in Gemm() local
427 const bool transpose_a = kLhsOrder == MapOrder::RowMajor; in test_gemm_impl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/ |
| H A D | linear_operator_util.py | 385 transpose_a=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 | 247 bool transpose_a = transpose_a_vals[i]; in TEST_P() local
|
| /aosp_15_r20/frameworks/rs/cpu_ref/ |
| H A D | rsCpuIntrinsicBLAS.cpp | 884 bool transpose_a = true; in kernelBNNM() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes.cc | 4478 bool transpose_a, in ConvertFullyConnectedImpl()
4587 bool transpose_a, in ConvertMatMulImpl()
4661 TRT_TensorOrWeights input_b, bool transpose_a, in ConvertMatMulHelper()
4681 bool transpose_a = false, transpose_b = false; in ConvertMatMul() local
4707 bool transpose_a = false, transpose_b = false; in ConvertBatchMatMul() local
|