Searched refs:blas_transpose_a (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_matmul_op_impl.h | 303 auto blas_transpose_a = trans[adj_x];
347 blas_transpose_a != se::blas::Transpose::kConjugateTranspose) {
353 auto gemv_trans_a = blas_transpose_a == se::blas::Transpose::kTranspose
372 ->ThenBlasGemm(blas_transpose_b, blas_transpose_a, n, m, k,
389 blas_transpose_b, blas_transpose_a, n, m, k,
420 auto blas_transpose_a = trans[adj_x];
465 ->ThenBlasGemm(blas_transpose_b, blas_transpose_a, n, m, k,
481 blas_transpose_b, blas_transpose_a, n, m, k,
|
| D | matmul_op.cc | 271 auto blas_transpose_a = trans[transpose_a]; in launch() local
311 blas_transpose_b, blas_transpose_a, n, m, k, alpha, b_ptr, in launch()
329 blas_transpose_b, blas_transpose_a, n, m, k, 1.0, b_ptr, in launch()
370 blas_transpose_b, blas_transpose_a, n, m, k, alpha, b_ptr, in launch()
411 ->ThenBlasGemm(blas_transpose_b, blas_transpose_a, n, m, k, in launch()
|