/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | bidirectional_sequence_rnn.cc | 129 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 D | bidirectional_sequence_lstm.cc | 489 const TfLiteTensor* aux_input = in Prepare() local 1125 const TfLiteTensor* aux_input = in Eval() local
|
H A D | lstm_eval.cc | 183 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 D | lstm.cc | 42 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/ |
D | bidirectional_sequence_rnn_1_3.mod.py | 40 bw_hidden_state, aux_input, fw_aux_weights, bw_aux_weights, activation, argument
|
D | bidirectional_sequence_rnn_state_output.mod.py | 43 bw_weights, bw_recurrent_weights, bw_bias, bw_hidden_state, aux_input, argument
|
D | bidirectional_sequence_lstm_state_output.mod.py | 116 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/ |
D | bidirectional_sequence_rnn.mod.py | 40 bw_hidden_state, aux_input, fw_aux_weights, bw_aux_weights, activation, argument
|
D | bidirectional_sequence_lstm_float16_batch_major_merge_outputs.mod.py | 117 aux_input = Input("input", "TENSOR_FLOAT16", "{{{}, {}, {}}}".format(n_batch, max_time, n_input)) variable
|
D | bidirectional_sequence_lstm_merge_outputs.mod.py | 117 aux_input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input)) variable
|
D | bidirectional_sequence_lstm_norm_fw_output.mod.py | 117 aux_input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input)) variable
|
D | bidirectional_sequence_lstm_float16_batch_major.mod.py | 116 aux_input = Input("input", "TENSOR_FLOAT16", "{{{}, {}, {}}}".format(n_batch, max_time, n_input)) variable
|
D | bidirectional_sequence_lstm_cifg_peephole.mod.py | 116 aux_input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input)) variable
|
D | bidirectional_sequence_lstm_aux_input.mod.py | 118 aux_input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input)) variable
|
D | bidirectional_sequence_lstm.mod.py | 116 aux_input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input)) variable
|
D | bidirectional_sequence_lstm_float16_batch_major_aux_input.mod.py | 119 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 D | lstm.cc | 356 const TfLiteTensor* aux_input, in EvalFloat()
|