| /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
|
| D | math_grad.py | 1714 def _SparseMatMul(t1, t2, out_dtype, transpose_a=False, transpose_b=False): argument
|
| D | math_ops.py | 2740 transpose_b=False, argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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
185 const bool transpose_b = false; in TEST_F() local
274 const bool transpose_b = false; in TEST_F() local
|
| D | matmul_op_test.cc | 100 bool transpose_b, Tensor* output, in RunMatMulWithBias()
119 bool transpose_a, bool transpose_b, const string& activation_type, in RunMatMulWithBiasAndActivation()
149 bool transpose_b, Tensor* output, in RunFusedMatMulOp()
218 bool transpose_b) { in VerifyMatMulWithBias()
239 bool transpose_b, 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 | 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()
|
| D | mkl_matmul_op.cc | 97 bool transpose_b = dim_pair[0].second == 1; in Compute() local
|
| D | meta_support.cc | 221 bool transpose_b, const quint8* a_data, const quint8* b_data, in QuantizedGemm()
|
| D | matmul_op.cc | 273 bool transpose_b = dim_pair[0].second == 1; in launch() local
|
| D | quantized_conv_ops.cc | 377 const bool transpose_b = 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 | 62 transpose_b, adjoint_a, argument
|
| D | csr_sparse_matrix_dense_mat_mul_grad_test.py | 62 transpose_b, adjoint_a, adjoint_b, argument
|
| /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
150 transpose_b=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 | 246 bool transpose_b = false; in __anond96a6a670702() local
409 bool transpose_b = false; in __anond96a6a670a02() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | benchmarks_test.py | 74 transpose_b=False, argument
412 def _benchmark_np_matmul(self, m, transpose_b, num_iters): argument
418 def _benchmark_tf_matmul(self, m, transpose_b, num_iters, argument
423 def _benchmark_gen_math_ops_matmul(self, m, transpose_b, num_iters): argument
430 def _benchmark_tfe_py_fastpath_execute_matmul(self, m, transpose_b, argument
438 def _benchmark_tfe_py_execute_matmul(self, m, transpose_b, num_iters): argument
454 transpose_b, argument
461 def _benchmark_nested_defun_matmul(self, m, transpose_b, num_iters): argument
465 def outer(a, b, c, transpose_b): argument
476 transpose_b, argument
|
| /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 | 379 transpose_b=False, argument
|
| D | linear_operator.py | 1180 transpose_b=False, argument
|
| /external/gemmlowp/test/ |
| 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
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 821 bool transpose_b = false; in CountMatMulOperations() local
962 AttrValue transpose_b; in CountBatchMatMulOperations() local
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | testlib.cc | 160 bool transpose_b) { in Matmul()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 4819 bool transpose_b, const string& node_name) { in ConvertFullyConnectedHelper()
4860 bool transpose_b, string node_name) { in ConvertMatMulHelper()
4959 bool transpose_b = attrs.get<bool>("transpose_b"); in ConvertMatMul() local
4990 const bool transpose_b = attrs.get<bool>("adj_y"); in ConvertBatchMatMul() local
|