Home
last modified time | relevance | path

Searched refs:kLower (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_activation_function_into_constants.cc94 static constexpr float kLower = 0; in Run() local
95 new_value = value < kLower ? kLower : value; in Run()
100 static constexpr float kLower = -1; in Run() local
101 new_value = value > kUpper ? kUpper : value < kLower ? kLower : value; in Run()
106 static constexpr float kLower = 0; in Run() local
107 new_value = value > kUpper ? kUpper : value < kLower ? kLower : value; in Run()
Dresolve_constant_unary.cc323 static constexpr float kLower = 0; in Run() local
324 new_value = value < kLower ? kLower : value; in Run()
329 static constexpr float kLower = -1; in Run() local
330 new_value = value > kUpper ? kUpper : value < kLower ? kLower : value; in Run()
335 static constexpr float kLower = 0; in Run() local
336 new_value = value > kUpper ? kUpper : value < kLower ? kLower : value; in Run()
/external/tensorflow/tensorflow/stream_executor/
Dblas.cc40 case UpperLower::kLower: in UpperLowerString()
Dblas.h73 enum class UpperLower { kUpper, kLower }; enumerator
/external/tensorflow/tensorflow/core/kernels/
Dtopk_op_gpu.h53 enum class PreferIndices { kLower, kHigher };
123 if (preferIndices == PreferIndices::kLower) {
311 auto max_heap = IndexedHeap<HeapType::kMaxHeap, PreferIndices::kLower,
Dmatrix_triangular_solve_op.cc210 upper_lower_matrix = se::blas::UpperLower::kLower; in ComputeMatrix()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcholesky_thunk.cc41 uplo_(options.lower() ? se::blas::UpperLower::kLower in CholeskyThunk()
Dcusolver_context.cc48 case se::blas::UpperLower::kLower: in CUDABlasUpperLower()
Dtriangular_solve_thunk.cc41 uplo_(options.lower() ? se::blas::UpperLower::kLower in TriangularSolveThunk()
Dcusolver_rewriter.cc67 se::blas::UpperLower uplo = options.lower() ? se::blas::UpperLower::kLower in CreateCholesky()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc350 case blas::UpperLower::kLower: in ROCMBlasUpperLower()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc290 case blas::UpperLower::kLower: in CUDABlasUpperLower()