Searched refs:TensorCuBlasGemm (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | blas_gemm.cc | 39 void TensorCuBlasGemm<T>::operator()(OpKernelContext* ctx, bool transa, in operator ()()
64 template struct TensorCuBlasGemm<Eigen::half>; variable
65 template struct TensorCuBlasGemm<float>; variable
|
| D | blas_gemm.h | 33 struct TensorCuBlasGemm { struct
64 TensorCuBlasGemm<T>()(ctx, transb, transa, n, m, k, alpha, b.data(),
|