HomeSort by relevance Sort by last modified time
    Searched refs:input_to_output_weights (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/lite/kernels/
lstm_eval.h 30 const TfLiteTensor* input_to_output_weights,
59 const TfLiteTensor* input_to_output_weights,
unidirectional_sequence_lstm.cc 270 const TfLiteTensor* input_to_output_weights = local
272 const int n_cell = input_to_output_weights->dims->data[0];
273 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->size, 2);
274 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->data[1], n_input);
309 const bool is_hybrid_op = ((input_to_output_weights->type == kTfLiteUInt8 ||
310 input_to_output_weights->type == kTfLiteInt8) &&
348 input_quantized->type = input_to_output_weights->type;
359 activation_state_quantized->type = input_to_output_weights->type;
373 cell_state_quantized->type = input_to_output_weights->type;
450 const TfLiteTensor* input_to_output_weights local
    [all...]
lstm.cc 344 const TfLiteTensor* input_to_output_weights = local
346 const int n_cell = input_to_output_weights->dims->data[0];
347 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->size, 2);
348 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->data[1], n_input);
385 const bool is_hybrid_op = ((input_to_output_weights->type == kTfLiteUInt8 ||
386 input_to_output_weights->type == kTfLiteInt8) &&
422 input_quantized->type = input_to_output_weights->type;
432 activation_state_quantized->type = input_to_output_weights->type;
445 cell_state_quantized->type = input_to_output_weights->type;
518 const TfLiteTensor* input_to_output_weights local
    [all...]
lstm_eval.cc 52 // input_to_output_weights
    [all...]
bidirectional_sequence_lstm.cc 204 const TfLiteTensor* input_to_output_weights = local
206 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->size, 2);
207 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->data[0], n_cell);
208 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->data[1], n_input);
209 TF_LITE_ENSURE_EQ(context, input_to_output_weights->type,
    [all...]

Completed in 123 milliseconds