| /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_grad.py | 1184 def _SparseMatMul(t1, t2, out_dtype, transpose_a=False, transpose_b=False): argument
|
| D | math_ops.py | 2406 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 | 24 static Graph* Matmul(int m, int k, int n, bool transpose_a, bool transpose_b, in Matmul()
|
| 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 | reference_gemm.h | 34 void ReferenceGemm(bool transpose_a, bool transpose_b, bool transpose_c, in ReferenceGemm()
|
| D | mkl_matmul_op.cc | 96 bool transpose_b = dim_pair[0].second == 1; in Compute() local
|
| D | matmul_op.cc | 270 bool transpose_b = dim_pair[0].second == 1; in launch() local
|
| D | meta_support.cc | 221 bool transpose_b, const quint8* a_data, const quint8* b_data, in QuantizedGemm()
|
| D | quantized_conv_ops.cc | 377 const bool transpose_b = false; in operator ()() local
|
| D | sparse_matmul_op.cc | 1028 bool transpose_b = transpose_b_; in Compute() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | benchmarks_test.py | 62 transpose_b=False, argument
324 def _benchmark_np_matmul(self, m, transpose_b, num_iters): argument
330 def _benchmark_tf_matmul(self, m, transpose_b, num_iters, argument
335 def _benchmark_gen_math_ops_matmul(self, m, transpose_b, num_iters): argument
342 def _benchmark_tfe_py_fastpath_execute_matmul(self, m, transpose_b, argument
350 def _benchmark_tfe_py_execute_matmul(self, m, transpose_b, num_iters): argument
367 transpose_b, argument
374 def _benchmark_nested_defun_matmul(self, m, transpose_b, num_iters): argument
378 def outer(a, b, c, transpose_b): argument
389 transpose_b, argument
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_util.py | 230 transpose_b=False, argument
415 transpose_b=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 | 75 const Tensor& transpose_b = context->input(4); in Compute() local
|
| /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 | 702 bool transpose_b = false; in CountMatMulOperations() local
834 AttrValue transpose_b; in CountBatchMatMulOperations() local
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | testlib.cc | 160 bool transpose_b) { in Matmul()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 220 bool transpose_a, transpose_b; in MatMulShape() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3840 bool transpose_b = attrs.get<bool>("transpose_b"); in ConvertMatMul() local
3872 const bool transpose_b = attrs.get<bool>("adj_y"); in ConvertBatchMatMul() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 2249 bool transpose_b = random_bool(generator()); in TEST_F() local
3063 bool transpose_b = random_bool(generator()); in TEST_F() local
|