| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_test.cc | 37 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel() argument
44 n_input_(n_input), in LSTMOpModel()
271 HybridLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in HybridLSTMOpModel() argument
277 : LSTMOpModel(n_batch, n_input, n_cell, n_output, use_cifg, use_peephole, in HybridLSTMOpModel()
451 const int n_input = 2; in TEST_F() local
456 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_F()
462 {n_batch, n_input}, // input tensor in TEST_F()
464 {n_cell, n_input}, // input_to_input_weight tensor in TEST_F()
465 {n_cell, n_input}, // input_to_forget_weight tensor in TEST_F()
466 {n_cell, n_input}, // input_to_cell_weight tensor in TEST_F()
[all …]
|
| D | bidirectional_sequence_lstm_test.cc | 37 BidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in BidirectionalLSTMOpModel() argument
45 n_input_(n_input), in BidirectionalLSTMOpModel()
407 const int n_input = 2; in TEST_P() local
415 n_batch, n_input, n_cell, n_output, sequence_length, /*use_cifg=*/false, in TEST_P()
421 {sequence_length, n_batch, n_input}, // input tensor in TEST_P()
424 {n_cell, n_input}, // input_to_input_weight tensor in TEST_P()
425 {n_cell, n_input}, // input_to_forget_weight tensor in TEST_P()
426 {n_cell, n_input}, // input_to_cell_weight tensor in TEST_P()
427 {n_cell, n_input}, // input_to_output_weight tensor in TEST_P()
447 {n_cell, n_input}, // input_to_input_weight tensor in TEST_P()
[all …]
|
| D | unidirectional_sequence_lstm_test.cc | 34 UnidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in UnidirectionalLSTMOpModel() argument
42 n_input_(n_input), in UnidirectionalLSTMOpModel()
243 int n_batch, int n_input, int n_cell, int n_output, int sequence_length, in HybridUnidirectionalLSTMOpModel() argument
249 n_batch, n_input, n_cell, n_output, sequence_length, time_major, in HybridUnidirectionalLSTMOpModel()
452 const int n_input = 2; in TEST_F() local
459 n_batch, n_input, n_cell, n_output, sequence_length, in TEST_F()
465 {sequence_length, n_batch, n_input}, // input tensor in TEST_F()
467 {n_cell, n_input}, // input_to_input_weight tensor in TEST_F()
468 {n_cell, n_input}, // input_to_forget_weight tensor in TEST_F()
469 {n_cell, n_input}, // input_to_cell_weight tensor in TEST_F()
[all …]
|
| D | optional_tensor_test.cc | 33 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel() argument
38 n_input_(n_input), in LSTMOpModel()
232 const int n_input = 2; in TEST() local
237 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST()
243 {n_batch, n_input}, // input tensor in TEST()
246 {n_cell, n_input}, // input_to_forget_weight tensor in TEST()
247 {n_cell, n_input}, // input_to_cell_weight tensor in TEST()
248 {n_cell, n_input}, // input_to_output_weight tensor in TEST()
|
| D | lstm_eval.cc | 116 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepWithAuxInput() argument
153 input_to_input_weights_ptr, n_cell, n_input, input_ptr_batch, n_batch, in LstmStepWithAuxInput()
158 input_to_forget_weights_ptr, n_cell, n_input, input_ptr_batch, n_batch, in LstmStepWithAuxInput()
161 input_to_cell_weights_ptr, n_cell, n_input, input_ptr_batch, n_batch, in LstmStepWithAuxInput()
164 input_to_output_weights_ptr, n_cell, n_input, input_ptr_batch, n_batch, in LstmStepWithAuxInput()
467 int n_batch, int n_cell, int n_input, int n_aux_input, int n_output, in LstmStepWithAuxInput()
503 if (!tensor_utils::IsZeroVector(input_ptr_batch, n_batch * n_input)) { in LstmStepWithAuxInput()
507 const int offset = b * n_input; in LstmStepWithAuxInput()
509 input_ptr_batch + offset, n_input, quantized_input_ptr_batch + offset, in LstmStepWithAuxInput()
519 input_to_input_weights_ptr, n_cell, n_input, in LstmStepWithAuxInput()
[all …]
|
| D | bidirectional_sequence_lstm.cc | 158 TfLiteContext* context, TfLiteNode* node, int n_input, int n_output, in CheckLstmTensorDimensionsAndTypes() argument
182 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[1], n_input); in CheckLstmTensorDimensionsAndTypes()
191 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[1], n_input); in CheckLstmTensorDimensionsAndTypes()
200 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[1], n_input); in CheckLstmTensorDimensionsAndTypes()
208 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->data[1], n_input); in CheckLstmTensorDimensionsAndTypes()
350 TfLiteNode* node, int n_input, in CheckInputTensorDimensions() argument
355 context, node, n_input, n_output, n_cell, in CheckInputTensorDimensions()
369 context, node, n_input, n_output, n_cell, in CheckInputTensorDimensions()
404 const int n_input = input->dims->data[2]; in Prepare() local
411 n_input); in Prepare()
[all …]
|
| D | unidirectional_sequence_lstm.cc | 101 TfLiteNode* node, int n_input, in CheckInputTensorDimensions() argument
116 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
123 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
129 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
268 const int n_input = input->dims->data[2]; in Prepare() local
274 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->data[1], n_input); in Prepare()
284 TF_LITE_ENSURE_OK(context, CheckInputTensorDimensions(context, node, n_input, in Prepare()
|
| D | lstm.cc | 117 TfLiteNode* node, int n_input, in CheckInputTensorDimensions() argument
134 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
141 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
147 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
342 const int n_input = input->dims->data[1]; in Prepare() local
348 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->data[1], n_input); in Prepare()
359 CheckInputTensorDimensions(context, node, n_input, n_output, in Prepare()
|
| /external/tensorflow/tensorflow/lite/experimental/microfrontend/ |
| D | audio_microfrontend_test.cc | 39 MicroFrontendOpModel(int n_input, int n_frame, int n_frequency_per_frame, in MicroFrontendOpModel() argument
43 : n_input_(n_input), in MicroFrontendOpModel()
180 const int n_input = 36; in TEST_F() local
184 MicroFrontendOpModel micro_frontend(n_input, n_frame, n_frequency_per_frame, in TEST_F()
187 {n_input}, in TEST_F()
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | unidirectional_sequence_rnn_test.py | 46 self.n_input = 28
76 "float", [None, self.time_steps, self.n_input], name="INPUT_IMAGE")
113 self.n_input))
148 sample_input = np.reshape(b1, (1, self.time_steps, self.n_input))
159 "float", [1, self.time_steps, self.n_input], name="INPUT_IMAGE_LITE")
|
| D | unidirectional_sequence_lstm_test.py | 46 self.n_input = 28
80 "float", [None, self.time_steps, self.n_input], name="INPUT_IMAGE")
118 self.n_input))
136 sample_input = np.reshape(b1, (1, self.time_steps, self.n_input))
147 "float", [1, self.time_steps, self.n_input], name="INPUT_IMAGE_LITE")
|
| D | bidirectional_sequence_lstm_test.py | 47 self.n_input = 28
81 "float", [None, self.time_steps, self.n_input], name="INPUT_IMAGE")
127 self.n_input))
147 sample_input = np.reshape(b1, (1, self.time_steps, self.n_input))
161 "float", [1, self.time_steps, self.n_input], name="INPUT_IMAGE_LITE")
|
| D | bidirectional_sequence_rnn_test.py | 48 self.n_input = 28
86 "float", [batch_size, self.time_steps, self.n_input],
146 self.n_input))
171 sample_input = np.reshape(b1, (1, self.time_steps, self.n_input))
185 "float", [1, self.time_steps, self.n_input], name="INPUT_IMAGE_LITE")
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fuse_quantized_convolution.cc | 76 int n_input = quantized_conv2D_node.input_size(); in FuseQuantizedConvolutionAndRequantize() local
79 n_input -= 1; // -1 since summand is moved after frozen min-max in FuseQuantizedConvolutionAndRequantize()
81 for (int i=0; i < n_input; i++) in FuseQuantizedConvolutionAndRequantize()
92 quantized_conv2D_node.input(n_input)]->input(0)]; in FuseQuantizedConvolutionAndRequantize()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 2259 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel() argument
2265 n_input_(n_input), in LSTMOpModel()
2569 const int n_input = 2; in TEST_F() local
2574 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_F()
2580 {n_batch, n_input}, // input tensor in TEST_F()
2582 {n_cell, n_input}, // input_to_input_weight tensor in TEST_F()
2583 {n_cell, n_input}, // input_to_forget_weight tensor in TEST_F()
2584 {n_cell, n_input}, // input_to_cell_weight tensor in TEST_F()
2585 {n_cell, n_input}, // input_to_output_weight tensor in TEST_F()
2673 const int n_input = 2; in TEST_F() local
[all …]
|
| /external/libxml2/ |
| D | testapi.c | 9458 int n_input; in test_xmlEncodeEntitiesReentrant() local
9461 for (n_input = 0;n_input < gen_nb_const_xmlChar_ptr;n_input++) { in test_xmlEncodeEntitiesReentrant()
9464 input = gen_const_xmlChar_ptr(n_input, 1); in test_xmlEncodeEntitiesReentrant()
9470 des_const_xmlChar_ptr(n_input, (const xmlChar *)input, 1); in test_xmlEncodeEntitiesReentrant()
9477 printf(" %d", n_input); in test_xmlEncodeEntitiesReentrant()
9504 int n_input; in test_xmlEncodeSpecialChars() local
9507 for (n_input = 0;n_input < gen_nb_const_xmlChar_ptr;n_input++) { in test_xmlEncodeSpecialChars()
9510 input = gen_const_xmlChar_ptr(n_input, 1); in test_xmlEncodeSpecialChars()
9516 des_const_xmlChar_ptr(n_input, (const xmlChar *)input, 1); in test_xmlEncodeSpecialChars()
9523 printf(" %d", n_input); in test_xmlEncodeSpecialChars()
[all …]
|