Home
last modified time | relevance | path

Searched refs:AsBlas (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc157 if (auto* blas = stream_exec->AsBlas()) { in LogConvAutotuneResults()
204 if (auto* blas = stream_exec->AsBlas()) { in LogFusedConvForwardAutotuneResults()
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc333 blas::BlasSupport *blas_support = AsBlas(); in GetBlasGemmAlgorithms()
343 blas::BlasSupport *blas_support = AsBlas(); in CreateBlasLtMatmulPlan()
355 blas::BlasSupport *blas_support = AsBlas(); in GetBlasLtMatmulAlgorithms()
434 blas::BlasSupport *StreamExecutor::AsBlas() { in AsBlas() function in stream_executor::StreamExecutor
Dstream_executor_pimpl.h507 blas::BlasSupport *AsBlas();
Dstream.cc1452 if (blas::BlasSupport *blas = stream->parent_->AsBlas()) { in Run()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_algorithm_picker.cc441 if (auto* blas = stream_exec_->AsBlas()) { in PickBestAlgorithmNoCacheCuda()