Searched refs:aux_input_to_input_weights (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | LSTM.h | 118 const float* aux_input_to_input_weights, const float* aux_input_to_forget_weights,
146 const _Float16* aux_input_to_input_weights, const _Float16* aux_input_to_forget_weights,
171 const float* aux_input_to_input_weights, const float* aux_input_to_forget_weights,
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_eval.h | 115 const TfLiteTensor* aux_input_to_input_weights,
151 const TfLiteTensor* aux_input_to_input_weights,
|
| H A D | lstm_eval.cc | 1762 const TfLiteTensor* aux_input_to_input_weights, in EvalFloat() argument
1837 GetTensorData<float>(aux_input_to_input_weights), in EvalFloat()
1899 GetTensorData<float>(aux_input_to_input_weights), in EvalFloat()
1957 const TfLiteTensor* aux_input_to_input_weights, in EvalHybrid()
2055 GetTensorData<int8_t>(aux_input_to_input_weights), in EvalHybrid()
2056 GetTensorScale(aux_input_to_input_weights), in EvalHybrid()
2153 GetTensorData<int8_t>(aux_input_to_input_weights), in EvalHybrid()
2154 GetTensorScale(aux_input_to_input_weights), in EvalHybrid()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 357 const TfLiteTensor* aux_input_to_input_weights, in EvalFloat() argument
431 GetTensorData<float>(aux_input_to_input_weights), in EvalFloat()
494 GetTensorData<float>(aux_input_to_input_weights), in EvalFloat()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 281 const TfLiteTensor* aux_input_to_input_weights, in EvalCalibration() argument
354 GetTensorData<float>(aux_input_to_input_weights), in EvalCalibration()
416 GetTensorData<float>(aux_input_to_input_weights), in EvalCalibration()
|