Home
last modified time | relevance | path

Searched defs:transpose_b (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dquantized_matmul_op_test.cc95 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
Dmatmul_op_test.cc102 bool transpose_b, Tensor* output, in RunMatMulWithBias()
121 bool transpose_a, bool transpose_b, const string& activation_type, in RunMatMulWithBiasAndActivation()
153 bool transpose_b, Tensor* output, in RunFusedMatMulOp()
222 bool transpose_b) { in VerifyMatMulWithBias()
243 bool transpose_b, in VerifyConv2DWithBiasAndActivation()
350 static Graph* Matmul(int m, int k, int n, bool transpose_a, bool transpose_b, in Matmul()
Dsparse_matmul_op_test.cc52 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()
Dreference_gemm.h34 void ReferenceGemm(bool transpose_a, bool transpose_b, bool transpose_c, in ReferenceGemm()
Dmeta_support.cc221 bool transpose_b, const quint8* a_data, const quint8* b_data, in QuantizedGemm()
Dquantized_conv_ops.cc377 const bool transpose_b = false; in operator ()() local
/external/tensorflow/tensorflow/python/ops/
Dmatmul_benchmark.py35 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/python/kernel_tests/linalg/sparse/
Dcsr_sparse_matrix_test.py151 def _testSparseSparse(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
191 def _testSparseDense(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
229 def _testDenseSparse(self, transpose_a, transpose_b, adjoint_a, adjoint_b): argument
Dcsr_sparse_matrix_sparse_mat_mul_grad_test.py62 transpose_b, adjoint_a, argument
Dcsr_sparse_matrix_dense_mat_mul_grad_test.py62 transpose_b, adjoint_a, adjoint_b, argument
/external/tensorflow/tensorflow/python/ops/linalg/sparse/
Dsparse_csr_matrix_ops.py121 def matmul_shape_inference(a, b, c, transpose_a, transpose_b, adjoint_a, argument
150 transpose_b=False, argument
/external/tensorflow/tensorflow/core/ops/
Dsparse_csr_matrix_ops_test.cc109 bool adjoint_b, bool transpose_output) { in TEST()
233 bool adjoint_b) { in TEST()
Dsparse_csr_matrix_ops.cc246 bool transpose_b = false; in __anon59a9d6c30702() local
409 bool transpose_b = false; in __anon59a9d6c30a02() local
/external/tensorflow/tensorflow/python/eager/
Dbenchmarks_test.py77 transpose_b=False, argument
410 def _benchmark_np_matmul(self, m, transpose_b, num_iters): argument
416 def _benchmark_tf_matmul(self, m, transpose_b, num_iters, argument
421 def _benchmark_gen_math_ops_matmul(self, m, transpose_b, num_iters): argument
428 def _benchmark_tfe_py_fastpath_execute_matmul(self, m, transpose_b, argument
436 def _benchmark_tfe_py_execute_matmul(self, m, transpose_b, num_iters): argument
452 transpose_b, argument
495 def _benchmark_nested_defun_matmul(self, m, transpose_b, num_iters): argument
499 def outer(a, b, c, transpose_b): argument
510 transpose_b, argument
/external/gemmlowp/eight_bit_int_gemm/
Deight_bit_int_gemm.cc217 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/core/kernels/mkl/
Dmkl_matmul_op.cc93 bool transpose_b = dim_pair[0].second == 1; in Compute() local
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_util.py389 transpose_b=False, argument
Dlinear_operator.py1215 transpose_b=False, argument
/external/tensorflow/tensorflow/c/experimental/ops/
Dmath_ops.cc92 bool transpose_a = false, bool transpose_b = false) { in MatMul()
/external/gemmlowp/test/
Dtest.cc35 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/python/ops/risc/
Drisc_ops.py109 transpose_b=False, argument
/external/tensorflow/tensorflow/core/graph/
Dtestlib.cc160 bool transpose_b) { in Matmul()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc1055 bool transpose_b = false; in CountMatMulOperations() local
1320 AttrValue transpose_b; in CountBatchMatMulOperations() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc5420 bool transpose_b, const NodeDef& node_def) { in ConvertFullyConnectedHelper()
5462 bool transpose_b, const NodeDef& node_def) { in ConvertMatMulHelper()
5561 bool transpose_b = attrs.get<bool>("transpose_b"); in ConvertMatMul() local
5592 const bool transpose_b = attrs.get<bool>("adj_y"); in ConvertBatchMatMul() local
/external/tensorflow/tensorflow/core/framework/
Dop_kernel_test.cc1072 AttrValue transpose_a, transpose_b, attr_t; in BM_TraceString() local

12