Searched refs:aux_input_to_output_weights (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | LSTM.h | 119 const float* aux_input_to_cell_weights, const float* aux_input_to_output_weights,
147 const _Float16* aux_input_to_cell_weights, const _Float16* aux_input_to_output_weights,
172 const float* aux_input_to_cell_weights, const float* aux_input_to_output_weights,
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_eval.h | 118 const TfLiteTensor* aux_input_to_output_weights,
154 const TfLiteTensor* aux_input_to_output_weights,
|
| H A D | lstm_eval.cc | 1765 const TfLiteTensor* aux_input_to_output_weights, in EvalFloat() argument
1840 GetTensorData<float>(aux_input_to_output_weights), in EvalFloat()
1902 GetTensorData<float>(aux_input_to_output_weights), in EvalFloat()
1960 const TfLiteTensor* aux_input_to_output_weights, in EvalHybrid()
2061 GetTensorData<int8_t>(aux_input_to_output_weights), in EvalHybrid()
2062 GetTensorScale(aux_input_to_output_weights), in EvalHybrid()
2159 GetTensorData<int8_t>(aux_input_to_output_weights), in EvalHybrid()
2160 GetTensorScale(aux_input_to_output_weights), in EvalHybrid()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 360 const TfLiteTensor* aux_input_to_output_weights, in EvalFloat() argument
434 GetTensorData<float>(aux_input_to_output_weights), in EvalFloat()
497 GetTensorData<float>(aux_input_to_output_weights), in EvalFloat()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 284 const TfLiteTensor* aux_input_to_output_weights, in EvalCalibration() argument
357 GetTensorData<float>(aux_input_to_output_weights), in EvalCalibration()
419 GetTensorData<float>(aux_input_to_output_weights), in EvalCalibration()
|