Home
last modified time | relevance | path

Searched defs:projection_weights (Results 1 – 25 of 56) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dunidirectional_sequence_lstm.cc147 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local
586 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in CheckInputTensorDimensions() local
761 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in PopulatePrecomputedZPTimesWeightsWithBias() local
1157 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in Prepare() local
1286 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in Eval() local
H A Dlstm.cc228 const TfLiteTensor* projection_weights = in PopulateQuantizedLstmParams8x8_16() local
528 const TfLiteTensor* projection_weights = in PopulateQuantizedLstmParams8x8_8() local
1029 const TfLiteTensor* projection_weights = in CheckInputTensorDimensions() local
1195 const TfLiteTensor* projection_weights = in PopulatePrecomputedZPTimesWeightsWithBias() local
1605 const TfLiteTensor* projection_weights = in Prepare() local
1734 const TfLiteTensor* projection_weights = in Prepare() local
1894 const TfLiteTensor* projection_weights = in Eval() local
H A Dlstm_eval.cc305 const float* projection_weights, in CalculateLstmOutputFloat()
477 const int8_t* projection_weights, const uint8_t* projection_weights_ledger, in CalculateLstmOutputHybrid()
662 const int8_t* projection_weights, int32_t proj_scale_a, in CalculateLstmOutputInteger8x8_16()
768 const int16_t* output_gate, const int8_t* projection_weights, in CalculateLstmOutputInteger8x8_8()
1768 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias, in EvalFloat()
1932 TfLiteStatus EvalHybrid( in EvalHybrid()
2232 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias, in EvalInteger8x8_16()
2470 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias, in EvalInteger8x8_8()
/aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/common/
H A DLSTMParams.h98 LSTMParams &set_projection_params(const T *projection_weights, const T *projection_bias) in set_projection_params()
225 const T *projection_weights() const in projection_weights() function
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
Dlayer_norm_lstm.mod.py64 projection_weights = Input("projection_weights", "TENSOR_FLOAT32", variable
236 projection_weights = Input("projection_weights", "TENSOR_FLOAT32", variable
Dlstm3_state2_float16.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT16", "{%d,%d}" % (n_output, n_cell)) variable
Dlstm2_state_float16.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT16", "{0,0}") variable
Dlstm2_float16.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT16", "{0,0}") variable
Dlstm3_float16.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT16", "{%d,%d}" % (n_output, n_cell)) variable
Dlstm3_state3_float16.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT16", "{%d,%d}" % (n_output, n_cell)) variable
Dlstm2_state2_float16.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT16", "{0,0}") variable
Dlstm3_state_float16.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT16", "{%d,%d}" % (n_output, n_cell)) variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
H A Dlstm.cc126 const float* projection_weights, const float* projection_bias, in CalculateLstmOutputCalibration()
287 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias, in EvalCalibration()
552 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in lstm_eval() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
H A Dlstm.cc363 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias, in EvalFloat()
598 const TfLiteTensor* projection_weights = GetOptionalInputTensor( in lstm_eval() local
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_0/
Dlstm3_state.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT32", "{%d,%d}" % (n_output, n_cell)) variable
Dlstm2_state.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT32", "{0,0}") variable
Dlstm2.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT32", "{0,0}") variable
Dlstm3_state3.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT32", "{%d,%d}" % (n_output, n_cell)) variable
Dlstm2_state2.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT32", "{0,0}") variable
Dlstm3.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT32", "{%d,%d}" % (n_output, n_cell)) variable
Dlstm3_state2.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT32", "{%d,%d}" % (n_output, n_cell)) variable
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_1/
Dlstm2_state2_relaxed.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT32", "{0,0}") variable
Dlstm2_state_relaxed.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT32", "{0,0}") variable
Dlstm2_relaxed.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT32", "{0,0}") variable
Dlstm3_state_relaxed.mod.py48 projection_weights = Input("projection_weights", "TENSOR_FLOAT32", "{%d,%d}" % (n_output, n_cell)) variable

123