experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Create comment to element wise clipping.

Browse Source 
PiperOrigin-RevId: 271650323
This commit is contained in:
Jian Li 2019-09-27 14:39:30 -07:00 committed by TensorFlower Gardener
parent be03ded52f
commit 5c271fd23d
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/lite/kernels/internal/tensor_utils.h
View File

@ -278,6 +278,12 @@ void CwiseMul(const int16_t* input_1, const int16_t* input_2,
void CwiseAdd(const int16_t* input_1, const int16_t* input_2, int n_batch,
int n_input, int16_t* output);
// Element-wise in-place clipping of a quantized vector.
// Parameters:
// - input: batch vector of size n_batch * n_input; 16 bit.
// - clipping_value: the value used for clipping.
// - n_batch: the number of batch.
// - n_input: the size for input and output.
void CwiseClipping(int16_t* input, const int16_t clipping_value,
int32_t n_batch, int32_t n_input);