Searched defs:abs_limit (Results 1 - 4 of 4) sorted by last modified time
/frameworks/ml/nn/common/operations/internal/optimized/ |
H A D | neon_tensor_utils.cc | 184 void NeonClipVector(const float* vector, int v_size, float abs_limit, argument 192 // Replicate abs_limit and -abs_limit in two vectors. 193 const float32x4_t abs_limit_f32x4 = vmovq_n_f32(abs_limit); 194 const float32x4_t neg_abs_limit_f32x4 = vmovq_n_f32(-abs_limit); 199 // Clip between abs_limit and -abs_limit. 207 result[v] = (abs_limit < vector[v]) ? abs_limit : vector[v]; 208 result[v] = (-abs_limit > resul [all...] |
H A D | neon_tensor_utils.h | 94 float Clip(float f, float abs_limit) { return PortableClip(f, abs_limit); } argument 96 void ClipVector(const float* vector, int v_size, float abs_limit, argument 98 NEON_OR_PORTABLE(ClipVector, vector, v_size, abs_limit, result);
|
/frameworks/ml/nn/common/operations/internal/reference/ |
H A D | portable_tensor_utils.cc | 24 float PortableClip(float f, float abs_limit) { argument 25 float result = (abs_limit < f) ? abs_limit : f; 26 result = (-abs_limit > result) ? -abs_limit : result; 139 void PortableClipVector(const float* vector, int v_size, float abs_limit, argument 142 *result++ = PortableClip(*vector++, abs_limit);
|
H A D | portable_tensor_utils.h | 26 // Limit a float input f betweeen +abs_limit and -abs_limit. 27 float PortableClip(float f, float abs_limit); 89 // Clip elements of a vector using a abs_limit value. 90 void PortableClipVector(const float* vector, int v_size, float abs_limit, 107 float Clip(float f, float abs_limit) { return PortableClip(f, abs_limit); } argument 174 void ClipVector(const float* vector, int v_size, float abs_limit, argument 176 PortableClipVector(vector, v_size, abs_limit, result);
|
Completed in 76 milliseconds