Searched refs:CwiseClipping (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/common/ |
| D | QuantUtils.h | 201 void CwiseClipping(int16_t* input, const int16_t clipping_value, int32_t n_batch, int32_t n_input);
203 void CwiseClipping(int8_t* input, const int8_t clipping_value, int32_t n_batch, int32_t n_input);
|
| D | QuantUtils.cpp | 150 void CwiseClipping(int16_t* input, const int16_t clipping_value, int32_t n_batch, int32_t n_input) { in CwiseClipping() function
164 void CwiseClipping(int8_t* input, const int8_t clipping_value, int32_t n_batch, int32_t n_input) { in CwiseClipping() function
|
| /packages/modules/NeuralNetworks/common/operations/ |
| D | QLSTM.cpp | 747 CwiseClipping(cellStateBuffer, quantizedCellClip, batchSize, numUnits); in execute()
786 CwiseClipping(outputBuffer, quantizedProjectionClip, batchSize, outputSize); in execute()
|
| D | LSTM.cpp | 942 tflite::tensor_utils::CwiseClipping(cell_state_out_buffer, n_batch * n_cell, in LSTMStep()
980 tflite::tensor_utils::CwiseClipping(output_buffer, n_batch * n_output, in LSTMStep()
|