Home
last modified time | relevance | path

Searched defs:aux_input (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbidirectional_sequence_rnn.cc129 const TfLiteTensor* aux_input = in Prepare() local
381 const TfLiteTensor* aux_input, in EvalFloat()
522 const TfLiteTensor* aux_input, const TfLiteTensor* aux_fw_input_weights, in EvalHybrid()
740 const TfLiteTensor* aux_input = in Eval() local
H A Dbidirectional_sequence_lstm.cc489 const TfLiteTensor* aux_input = in Prepare() local
1125 const TfLiteTensor* aux_input = in Eval() local
H A Dlstm_eval.cc183 const float* aux_input, const float* aux_input_to_gate_weights, in CalculateLstmGateFloat()
347 const int8_t* aux_input, const float* aux_input_sf, in CalculateLstmGateHybrid()
1761 const TfLiteTensor* aux_input, in EvalFloat()
1932 TfLiteStatus EvalHybrid( in EvalHybrid()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
H A Dlstm.cc42 const float* aux_input, const float* aux_input_to_gate_weights, in CalculateLstmGateFloat()
280 const TfLiteTensor* aux_input, in EvalCalibration()
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/
Dbidirectional_sequence_rnn_1_3.mod.py40 bw_hidden_state, aux_input, fw_aux_weights, bw_aux_weights, activation, argument
Dbidirectional_sequence_rnn_state_output.mod.py43 bw_weights, bw_recurrent_weights, bw_bias, bw_hidden_state, aux_input, argument
Dbidirectional_sequence_lstm_state_output.mod.py116 aux_input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input)) variable
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
Dbidirectional_sequence_rnn.mod.py40 bw_hidden_state, aux_input, fw_aux_weights, bw_aux_weights, activation, argument
Dbidirectional_sequence_lstm_float16_batch_major_merge_outputs.mod.py117 aux_input = Input("input", "TENSOR_FLOAT16", "{{{}, {}, {}}}".format(n_batch, max_time, n_input)) variable
Dbidirectional_sequence_lstm_merge_outputs.mod.py117 aux_input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input)) variable
Dbidirectional_sequence_lstm_norm_fw_output.mod.py117 aux_input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input)) variable
Dbidirectional_sequence_lstm_float16_batch_major.mod.py116 aux_input = Input("input", "TENSOR_FLOAT16", "{{{}, {}, {}}}".format(n_batch, max_time, n_input)) variable
Dbidirectional_sequence_lstm_cifg_peephole.mod.py116 aux_input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input)) variable
Dbidirectional_sequence_lstm_aux_input.mod.py118 aux_input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input)) variable
Dbidirectional_sequence_lstm.mod.py116 aux_input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input)) variable
Dbidirectional_sequence_lstm_float16_batch_major_aux_input.mod.py119 aux_input = Input("input", "TENSOR_FLOAT16", "{{{}, {}, {}}}".format(n_batch, max_time, n_input)) variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
H A Dlstm.cc356 const TfLiteTensor* aux_input, in EvalFloat()