HomeSort by relevance Sort by last modified time
    Searched refs:TensorCuBlasGemm (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/rnn/kernels/
blas_gemm.cc 39 void TensorCuBlasGemm<T>::operator()(OpKernelContext* ctx, bool transa,
64 template struct TensorCuBlasGemm<Eigen::half>;
65 template struct TensorCuBlasGemm<float>;
blas_gemm.h 33 struct TensorCuBlasGemm {
64 TensorCuBlasGemm<T>()(ctx, transb, transa, n, m, k, alpha, b.data(),

Completed in 189 milliseconds