Searched refs:input_to_input_weights (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm.cc | 111 const TfLiteTensor* input_to_input_weights = in CheckInputTensorDimensions() local
113 if (input_to_input_weights != nullptr) { in CheckInputTensorDimensions()
114 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->size, 2); in CheckInputTensorDimensions()
115 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
116 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
159 ((input_to_input_weights != nullptr) && in CheckInputTensorDimensions()
161 ((input_to_input_weights == nullptr) && in CheckInputTensorDimensions()
187 const bool use_cifg = (input_to_input_weights == nullptr); in CheckInputTensorDimensions()
326 const TfLiteTensor* input_to_input_weights = in Prepare() local
328 const bool use_cifg = (input_to_input_weights == nullptr); in Prepare()
[all …]
|
| D | lstm.cc | 128 const TfLiteTensor* input_to_input_weights = in CheckInputTensorDimensions() local
130 const bool use_cifg = (input_to_input_weights == nullptr); in CheckInputTensorDimensions()
132 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->size, 2); in CheckInputTensorDimensions()
133 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
134 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
177 ((input_to_input_weights != nullptr) && in CheckInputTensorDimensions()
179 ((input_to_input_weights == nullptr) && in CheckInputTensorDimensions()
402 const TfLiteTensor* input_to_input_weights = in Prepare() local
404 const bool use_cifg = (input_to_input_weights == nullptr); in Prepare()
512 const TfLiteTensor* input_to_input_weights = in Eval() local
[all …]
|
| D | lstm_eval.h | 27 const TfLiteTensor* input, const TfLiteTensor* input_to_input_weights,
56 const TfLiteTensor* input, const TfLiteTensor* input_to_input_weights,
|
| D | bidirectional_sequence_lstm.cc | 186 const TfLiteTensor* input_to_input_weights = in CheckLstmTensorDimensionsAndTypes() local
188 if (input_to_input_weights != nullptr) { in CheckLstmTensorDimensionsAndTypes()
189 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->size, 2); in CheckLstmTensorDimensionsAndTypes()
190 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
191 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[1], n_input); in CheckLstmTensorDimensionsAndTypes()
192 TF_LITE_ENSURE_EQ(context, input_to_input_weights->type, in CheckLstmTensorDimensionsAndTypes()
246 ((input_to_input_weights != nullptr) && in CheckLstmTensorDimensionsAndTypes()
248 ((input_to_input_weights == nullptr) && in CheckLstmTensorDimensionsAndTypes()
280 const bool use_cifg = (input_to_input_weights == nullptr); in CheckLstmTensorDimensionsAndTypes()
|
| D | lstm_eval.cc | 871 const TfLiteTensor* input, const TfLiteTensor* input_to_input_weights, in EvalFloat() argument
917 const bool use_cifg = (input_to_input_weights == nullptr); in EvalFloat()
939 (use_cifg) ? nullptr : input_to_input_weights->data.f; in EvalFloat()
1069 const TfLiteTensor* input, const TfLiteTensor* input_to_input_weights, in EvalHybrid() argument
1120 const bool use_cifg = (input_to_input_weights == nullptr); in EvalHybrid()
1147 GetInt8DataPtr(input_to_input_weights, is_uint8_hybrid); in EvalHybrid()
1151 input_to_input_weights_scale = input_to_input_weights->params.scale; in EvalHybrid()
|