Home
last modified time | relevance | path

Searched refs:kF32 (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dblas.cc79 case ComputationType::kF32: in ComputationTypeString()
Dblas.h102 kF32, // 32-bit floating-point enumerator
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc99 computation_type = se::blas::ComputationType::kF32; in DoGemmWithAlgorithm()
102 computation_type = se::blas::ComputationType::kF32; in DoGemmWithAlgorithm()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc403 case blas::ComputationType::kF32: in CUDAComputationType()
426 case blas::ComputationType::kF32: // fall-through in CUBLASComputationType()
449 case blas::ComputationType::kF32: // fall-through in GetScaleType()
2271 if (computation_type == blas::ComputationType::kF32) { in DoBlasInternalImpl()
3243 return blas::ComputationType::kF32; in DoBlasInternalImpl()