Home
last modified time | relevance | path

Searched refs:ThenBlasGemm (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/contrib/rnn/kernels/
Dblas_gemm.cc55 ->ThenBlasGemm(trans[transa], trans[transb], m, n, k, alpha, a_ptr, in operator ()()
/external/tensorflow/tensorflow/contrib/image/kernels/
Dadjust_hsv_in_yiq_op_gpu.cu.cc75 ->ThenBlasGemm(no_transpose, no_transpose, n, m, k, 1.0f, b_ptr, n, in operator ()()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_3d.cc258 ->ThenBlasGemm(no_transpose, no_transpose, n, m, k, 1.0f, b_ptr, in launch()
285 ->ThenBlasGemm(no_transpose, no_transpose, n, m, k, 1.0f, b_ptr, in launch()
Dbatch_matmul_op_impl.h372 ->ThenBlasGemm(blas_transpose_b, blas_transpose_a, n, m, k,
465 ->ThenBlasGemm(blas_transpose_b, blas_transpose_a, n, m, k,
Dconv_grad_ops_3d.cc1189 ->ThenBlasGemm(transpose, no_transpose, n, m, k, 1.0f, b_ptr, k, in Compute()
1218 ->ThenBlasGemm(transpose, no_transpose, n, m, k, 1.0f, b_ptr, k, in Compute()
1582 ->ThenBlasGemm(se::blas::Transpose::kNoTranspose, in Compute()
1609 ->ThenBlasGemm(se::blas::Transpose::kNoTranspose, in Compute()
Dconv_ops.cc629 ->ThenBlasGemm(no_transpose, no_transpose, n, m, k, 1.0f, b_ptr, n, in operator ()()
657 ->ThenBlasGemm(no_transpose, no_transpose, n, m, k, 1.0f, b_ptr, n, in operator ()()
Dconv_grad_filter_ops.cc637 ->ThenBlasGemm(se::blas::Transpose::kNoTranspose, in operator ()()
668 ->ThenBlasGemm(se::blas::Transpose::kNoTranspose, in operator ()()
Dconv_grad_input_ops.cc767 ->ThenBlasGemm(transpose, no_transpose, n, m, k, 1.0f, b_ptr, k, in operator ()()
800 ->ThenBlasGemm(transpose, no_transpose, n, m, k, 1.0f, b_ptr, k, in operator ()()
Dmatmul_op.cc411 ->ThenBlasGemm(blas_transpose_b, blas_transpose_a, n, m, k, in launch()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc74 ->ThenBlasGemm( in DoGemm()
/external/tensorflow/tensorflow/stream_executor/
Dstream.h1250 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1256 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1261 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1266 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
1274 TF_EXPORT Stream &ThenBlasGemm(blas::Transpose transa, blas::Transpose transb,
Dstream.cc3498 Stream &Stream::ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm() function in stream_executor::Stream
3516 Stream &Stream::ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm() function in stream_executor::Stream
3532 Stream &Stream::ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm() function in stream_executor::Stream
3548 Stream &Stream::ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm() function in stream_executor::Stream
3569 Stream &Stream::ThenBlasGemm(blas::Transpose transa, blas::Transpose transb, in ThenBlasGemm() function in stream_executor::Stream
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3572 stream->ThenBlasGemm(blas::Transpose::kNoTranspose, in DoMatMul()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3332 stream->ThenBlasGemm(blas::Transpose::kNoTranspose, in DoMatMul()