Searched defs:abs_limit (Results 1 - 4 of 4) sorted by last modified time

/frameworks/ml/nn/common/operations/internal/optimized/
H A Dneon_tensor_utils.cc184 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 Dneon_tensor_utils.h94 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 Dportable_tensor_utils.cc24 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 Dportable_tensor_utils.h26 // 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