Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dblas.cc77 case ComputationType::kF32: in ComputationTypeString()
Dblas.h98 kF32, // 32-bit floating-point enumerator
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc257 return se::blas::ComputationType::kF32; in GetBlasComputationType()
259 return se::blas::ComputationType::kF32; in GetBlasComputationType()
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op.cc226 *compute_type = ComputationType::kF32; in GetCublasAutotuneComputationType()
232 *compute_type = ComputationType::kF32; in GetCublasAutotuneComputationType()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc387 case blas::ComputationType::kF32: in CUDAComputationType()
2060 if (computation_type == blas::ComputationType::kF32) { in DoBlasGemmWithAlgorithm()