| /external/tensorflow/tensorflow/python/ops/ |
| D | matmul_benchmark_test.py | 32 def BuildGraphTest(n, m, k, transpose_a, transpose_b, dtype): argument
46 def RunGraphTest(n, m, k, transpose_a, transpose_b, dtype): argument
71 def _VerifyBuildGraph(self, n, m, k, transpose_a, transpose_b, dtype): argument
104 def _VerifyRunGraph(self, n, m, k, transpose_a, transpose_b, dtype): argument
|
| 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
|
| D | math_ops.py | 2405 transpose_a=False, argument
2586 transpose_a=False, argument
|
| D | math_grad.py | 1184 def _SparseMatMul(t1, t2, out_dtype, transpose_a=False, transpose_b=False): 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 | 24 static Graph* Matmul(int m, int k, int n, bool transpose_a, bool transpose_b, in Matmul()
|
| 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 | reference_gemm.h | 34 void ReferenceGemm(bool transpose_a, bool transpose_b, bool transpose_c, in ReferenceGemm()
|
| D | mkl_matmul_op.cc | 95 bool transpose_a = dim_pair[0].first == 0; in Compute() local
|
| D | matmul_op.cc | 269 bool transpose_a = dim_pair[0].first == 0; in launch() local
|
| D | meta_support.cc | 220 void QuantizedGemm(OpKernelContext* tf_context, bool transpose_a, in QuantizedGemm()
|
| D | quantized_conv_ops.cc | 376 const bool transpose_a = false; in operator ()() local
|
| /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()
351 bool transpose_a, bool conjugate_a, in BuildTriangularSolve()
483 bool transpose_a = in ExpandInstruction() local
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_util.py | 229 transpose_a=False, argument
414 transpose_a=False, argument
|
| /external/tensorflow/tensorflow/contrib/factorization/python/kernel_tests/ |
| D | masked_matmul_benchmark.py | 63 transpose_a=False, transpose_b=False): argument
|
| D | masked_matmul_ops_test.py | 60 def _runTestMaskedProduct(self, transpose_a, transpose_b): argument
|
| /external/gemmlowp/eight_bit_int_gemm/ |
| D | eight_bit_int_gemm.cc | 220 bool CanHandleMetaFastpath(bool transpose_a, bool transpose_b, bool transpose_c, in CanHandleMetaFastpath()
304 void EightBitIntGemm(bool transpose_a, bool transpose_b, bool transpose_c, in EightBitIntGemm()
341 void EightBitIntGemm(bool transpose_a, bool transpose_b, bool transpose_c, in EightBitIntGemm()
|
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| D | masked_matmul_ops.cc | 74 const Tensor& transpose_a = context->input(3); in Compute() local
|
| /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/xla/tests/ |
| D | triangular_solve_test.cc | 438 TriangularSolveOptions::Transpose transpose_a; member
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 566 Transpose transpose_a = 4; field
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model.cc | 188 const bool transpose_a = node.attr().at("transpose_a").b(); in CalculateFlops() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 701 bool transpose_a = false; in CountMatMulOperations() local
827 AttrValue transpose_a; in CountBatchMatMulOperations() local
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | testlib.cc | 159 Node* Matmul(Graph* g, Node* in0, Node* in1, bool transpose_a, in Matmul()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 748 int transpose_a) { in TriangularSolve()
|