Home
last modified time | relevance | path

Searched refs:blas_version (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc156 string blas_version; in LogConvAutotuneResults() local
158 if (blas->GetVersion(&blas_version).ok()) { in LogConvAutotuneResults()
159 log.set_blas_version(blas_version); in LogConvAutotuneResults()
203 string blas_version; in LogFusedConvForwardAutotuneResults() local
205 if (blas->GetVersion(&blas_version).ok()) { in LogFusedConvForwardAutotuneResults()
206 log.set_blas_version(blas_version); in LogFusedConvForwardAutotuneResults()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_algorithm_denylist.cc33 blas_version: "10201"
40 blas_version: "10201"
Dhlo_algorithm_denylist.h29 const std::string& blas_version, const std::string& hlo);
Dgpu_autotuning.proto27 string blas_version = 5; field
Dgpu_conv_algorithm_picker.cc440 string blas_version; in PickBestAlgorithmNoCacheCuda() local
442 (void)blas->GetVersion(&blas_version); in PickBestAlgorithmNoCacheCuda()
447 blas_version, canonical_hlo); in PickBestAlgorithmNoCacheCuda()
516 entry->set_blas_version(blas_version); in PickBestAlgorithmNoCacheCuda()
597 log.set_blas_version(blas_version); in PickBestAlgorithmNoCacheCuda()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/data/
Dhlo_algorithm_denylist.pbtxt5 blas_version: "9000"
13 blas_version: "9000"
/external/tensorflow/tensorflow/core/protobuf/
Dautotuning.proto85 string blas_version = 6; field