Combine PortableCwiseClipping in portable_tensor_utils from reference and fp16 implementations.
PiperOrigin-RevId: 351283702 Change-Id: I4333e5fed8740b6d684d2707953b25824e97c15a
This commit is contained in:
parent
741b16bd0f
commit
8f9ba797cc
@ -174,7 +174,7 @@ void PortableCwiseAdd(const int16_t* input_1, const int16_t* input_2,
|
|||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
void PortableCwiseClipping(T* vector, const int v_size,
|
void PortableCwiseClipping(T* vector, const int v_size,
|
||||||
const T clipping_value) {
|
const T& clipping_value) {
|
||||||
for (int i = 0; i < v_size; i++) {
|
for (int i = 0; i < v_size; i++) {
|
||||||
vector[i] = std::max(std::min(clipping_value, vector[i]),
|
vector[i] = std::max(std::min(clipping_value, vector[i]),
|
||||||
static_cast<T>(-clipping_value));
|
static_cast<T>(-clipping_value));
|
||||||
|
Loading…
x
Reference in New Issue
Block a user