| /external/tensorflow/tensorflow/python/ops/ |
| D | matmul_benchmark.py | 35 def build_graph(device, n, m, k, transpose_a, transpose_b, dtype): argument
71 def run_graph(self, device, n, m, k, transpose_a, transpose_b, num_iters, argument
118 def run_test_gpu(self, n, m, k, transpose_a, transpose_b, dtype, num_iters): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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
180 const bool transpose_a = true; in TEST_F() local
273 const bool transpose_a = true; in TEST_F() local
|
| D | matmul_op_test.cc | 99 const Tensor& bias_data, bool transpose_a, in RunMatMulWithBias()
119 bool transpose_a, bool transpose_b, const string& activation_type, in RunMatMulWithBiasAndActivation()
148 const std::vector<string>& fused_ops, bool transpose_a, in RunFusedMatMulOp()
217 void VerifyMatMulWithBias(int m, int k, int n, bool transpose_a, in VerifyMatMulWithBias()
238 void VerifyConv2DWithBiasAndActivation(int m, int k, int n, bool transpose_a, in VerifyConv2DWithBiasAndActivation()
346 static Graph* Matmul(int m, int k, int n, bool transpose_a, bool transpose_b, in Matmul()
|
| D | reference_gemm.h | 34 void ReferenceGemm(bool transpose_a, bool transpose_b, bool transpose_c, in ReferenceGemm()
|
| 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()
|
| D | mkl_matmul_op.cc | 96 bool transpose_a = dim_pair[0].first == 0; in Compute() local
|
| D | meta_support.cc | 220 void QuantizedGemm(OpKernelContext* tf_context, bool transpose_a, in QuantizedGemm()
|
| D | matmul_op.cc | 272 bool transpose_a = dim_pair[0].first == 0; in launch() local
|
| D | quantized_conv_ops.cc | 376 const bool transpose_a = false; in operator ()() local
|
| /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
194 def _testSparseDense(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
235 def _testDenseSparse(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
|
| D | csr_sparse_matrix_sparse_mat_mul_grad_test.py | 61 def _testLargeBatchSparseMatrixSparseMatMulGrad(self, datatype, transpose_a, argument
|
| D | csr_sparse_matrix_dense_mat_mul_grad_test.py | 61 def _testLargeBatchSparseMatrixMatMulGrad(self, datatype, transpose_a, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 124 XlaOp InvertDiagonalBlocks(XlaOp diag_blocks, bool lower, bool transpose_a, in InvertDiagonalBlocks()
254 bool transpose_a, bool conjugate_a, in SolveWithInvertedDiagonalBlocks()
368 bool transpose_a, bool conjugate_a, in BuildTriangularSolve()
500 bool transpose_a = in ExpandInstruction() local
|
| /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
149 transpose_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 | 245 bool transpose_a = false; in __anond96a6a670702() local
408 bool transpose_a = false; in __anond96a6a670a02() local
|
| /external/gemmlowp/eight_bit_int_gemm/ |
| 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()
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_util.py | 378 transpose_a=False, argument
|
| D | linear_operator.py | 1179 transpose_a=False, argument
|
| /external/gemmlowp/test/ |
| 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
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 464 auto transpose_a = in ImportInstruction() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | triangular_solve_test.cc | 446 TriangularSolveOptions::Transpose transpose_a; member
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 579 Transpose transpose_a = 4; field
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model.cc | 189 const bool transpose_a = node.attr().at("transpose_a").b(); in CalculateFlops() local
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 815 CSRSparseMatrixMatVec(bool transpose_a, bool conjugate_a) in CSRSparseMatrixMatVec()
|