Searched refs:ThenBlasLtMatmulImpl (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 4326 Stream &Stream::ThenBlasLtMatmulImpl( in ThenBlasLtMatmulImpl() function in stream_executor::Stream
4349 template Stream &Stream::ThenBlasLtMatmulImpl<int8, int32>(
4356 template Stream &Stream::ThenBlasLtMatmulImpl<Eigen::half, Eigen::half>(
4363 template Stream &Stream::ThenBlasLtMatmulImpl<float, float>(
4370 template Stream &Stream::ThenBlasLtMatmulImpl<double, double>(
4378 Stream::ThenBlasLtMatmulImpl<std::complex<float>, std::complex<float>>(
4389 Stream::ThenBlasLtMatmulImpl<std::complex<double>, std::complex<double>>(
|
| D | stream.h | 1663 return ThenBlasLtMatmulImpl(plan, alpha, a, b, beta, c, scratch_allocator,
2101 Stream &ThenBlasLtMatmulImpl(const blas::IBlasLtMatmulPlan *plan,
|