Searched refs:CwiseClipping (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/common/ |
| D | QuantUtils.h | 205 void CwiseClipping(int16_t* input, const int16_t clipping_value, int32_t n_batch, int32_t n_input);
207 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/cpu_operations/ |
| D | QLSTM.cpp | 631 CwiseClipping(cellStateBuffer, quantizedCellClip, batchSize, numUnits); in execute()
670 CwiseClipping(outputBuffer, quantizedProjectionClip, batchSize, outputSize); in execute()
|
| D | LSTM.cpp | 943 tflite::tensor_utils::CwiseClipping(cell_state_out_buffer, n_batch * n_cell, in LSTMStep()
982 tflite::tensor_utils::CwiseClipping(output_buffer, n_batch * n_output, in LSTMStep()
|