| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| H A D | lstm.cc | 36 // Input Tensors of size {n_batch, n_input}
39 // Input weight tensors of size: {n_cell, n_input}
75 TfLiteNode* node, int n_input,
90 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[1], n_input);
97 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[1], n_input);
103 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[1], n_input);
234 const int n_input = input->dims->data[1]; local
240 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->data[1], n_input);
250 CheckInputTensorDimensions(context, node, n_input, n_output, n_cell);
353 const int n_input local
74 CheckInputTensorDimensions(TfLiteContext* context, TfLiteNode* node, int n_input, int n_output, int n_cell) argument
[all...] |
| H A D | optional_tensor_test.cc | 33 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, argument
38 n_input_(n_input),
248 const int n_input = 2; local
253 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output,
259 {n_batch, n_input}, // input tensor
262 {n_cell, n_input}, // input_to_forget_weight tensor
263 {n_cell, n_input}, // input_to_cell_weight tensor
264 {n_cell, n_input}, // input_to_output_weight tensor
|
| H A D | unidirectional_sequence_lstm.cc | 36 // Input Tensors of size {max_time, n_batch, n_input}
39 // Input weight tensors of size: {n_cell, n_input}
75 TfLiteNode* node, int n_input,
90 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[1], n_input);
97 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[1], n_input);
103 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[1], n_input);
235 const int n_input = input->dims->data[2]; local
241 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->data[1], n_input);
251 CheckInputTensorDimensions(context, node, n_input, n_output, n_cell);
356 const int n_input local
74 CheckInputTensorDimensions(TfLiteContext* context, TfLiteNode* node, int n_input, int n_output, int n_cell) argument
[all...] |
| H A D | lstm_test.cc | 35 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, argument
40 n_input_(n_input),
246 const int n_input = 2; local
251 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output,
257 {n_batch, n_input}, // input tensor
259 {n_cell, n_input}, // input_to_input_weight tensor
260 {n_cell, n_input}, // input_to_forget_weight tensor
261 {n_cell, n_input}, // input_to_cell_weight tensor
262 {n_cell, n_input}, // input_to_output_weight tensor
356 const int n_input local
461 const int n_input = 5; local
[all...] |
| H A D | unidirectional_sequence_lstm_test.cc | 35 UnidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, argument
42 n_input_(n_input),
252 const int n_input = 2; local
259 n_batch, n_input, n_cell, n_output, sequence_length, /*use_cifg=*/false,
263 {sequence_length, n_batch, n_input}, // input tensor
265 {n_cell, n_input}, // input_to_input_weight tensor
266 {n_cell, n_input}, // input_to_forget_weight tensor
267 {n_cell, n_input}, // input_to_cell_weight tensor
268 {n_cell, n_input}, // input_to_output_weight tensor
332 // Input should have n_input * sequence_lengt
360 const int n_input = 2; local
462 const int n_input = 5; local
[all...] |
| /external/libxml2/ |
| H A D | testapi.c | 9458 int n_input; local
9461 for (n_input = 0;n_input < gen_nb_const_xmlChar_ptr;n_input++) {
9464 input = gen_const_xmlChar_ptr(n_input, 1);
9470 des_const_xmlChar_ptr(n_input, (const xmlChar *)input, 1);
9477 printf(" %d", n_input);
9504 int n_input; local
9507 for (n_input = 0;n_input < gen_nb_const_xmlChar_pt
13302 int n_input; local
13536 int n_input; local
16473 int n_input; local
16797 int n_input; local
29863 int n_input; local
29893 int n_input; local
30160 int n_input; local
32982 int n_input; local
35068 int n_input; local
[all...] |