Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmatmul_util.cc90 return use_f32_for_f16_computation ? ComputationType::kF32 in GetBlasComputationType()
93 return ComputationType::kF32; in GetBlasComputationType()
97 : ComputationType::kF32; in GetBlasComputationType()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_blas_utils.cc69 case blas::ComputationType::kF32: in AsCublasComputeType()
Dcuda_blas_lt.cc133 return ((computation_type == blas::ComputationType::kF32) && in GetScaleType()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dblas.cc79 case ComputationType::kF32: in ComputationTypeString()
Dblas.h107 kF32, // 32-bit floating-point enumerator
Dstream.h1511 case blas::ComputationType::kF32: in CheckTypesForExtendedBlas()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dmatmul_utils.cc391 return se::blas::ComputationType::kF32; in GetBlasComputationType()