Searched refs:blas_transpose_b (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_matmul_op_impl.h | 304 auto blas_transpose_b = trans[adj_y];
346 blas_transpose_b != se::blas::Transpose::kConjugateTranspose &&
372 ->ThenBlasGemm(blas_transpose_b, blas_transpose_a, n, m, k,
389 blas_transpose_b, blas_transpose_a, n, m, k,
421 auto blas_transpose_b = trans[adj_y];
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 | 272 auto blas_transpose_b = trans[transpose_b]; 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()
|