Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dblas.cc49 case Diagonal::kUnit: in DiagonalString()
Dblas.h83 enum class Diagonal { kUnit, kNonUnit }; enumerator
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtriangular_solve_thunk.cc45 unit_diagonal_(options.unit_diagonal() ? se::blas::Diagonal::kUnit in TriangularSolveThunk()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc403 case blas::Diagonal::kUnit: in ROCMBlasDiagonal()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc315 case blas::Diagonal::kUnit: in CUDABlasDiagonal()