Home
last modified time | relevance | path

Searched refs:n_cell (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_lstm_test.cc37 BidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in BidirectionalLSTMOpModel() argument
46 n_fw_cell_(n_cell), in BidirectionalLSTMOpModel()
47 n_bw_cell_(n_cell), in BidirectionalLSTMOpModel()
409 const int n_cell = 4; in TEST_P() local
415 n_batch, n_input, n_cell, n_output, sequence_length, /*use_cifg=*/false, 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()
429 {n_cell, n_output}, // recurrent_to_input_weight tensor in TEST_P()
[all …]
Dlstm_test.cc37 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel() argument
45 n_cell_(n_cell), 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()
453 const int n_cell = 4; in TEST_F() local
456 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, 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()
467 {n_cell, n_input}, // input_to_output_weight tensor in TEST_F()
[all …]
Dunidirectional_sequence_lstm_test.cc34 UnidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, in UnidirectionalLSTMOpModel() argument
43 n_cell_(n_cell), 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()
454 const int n_cell = 4; in TEST_F() local
459 n_batch, n_input, n_cell, n_output, sequence_length, 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()
470 {n_cell, n_input}, // input_to_output_weight tensor in TEST_F()
[all …]
Dlstm_eval.cc116 const TfLiteLSTMParams* params, int n_batch, int n_cell, int n_input, in LstmStepWithAuxInput() argument
132 tensor_utils::ZeroVector(input_gate_scratch, n_cell * n_batch); in LstmStepWithAuxInput()
134 tensor_utils::ZeroVector(forget_gate_scratch, n_cell * n_batch); in LstmStepWithAuxInput()
135 tensor_utils::ZeroVector(cell_scratch, n_cell * n_batch); in LstmStepWithAuxInput()
136 tensor_utils::ZeroVector(output_gate_scratch, n_cell * n_batch); in LstmStepWithAuxInput()
139 tensor_utils::VectorBatchVectorAssign(input_gate_bias_ptr, n_cell, in LstmStepWithAuxInput()
142 tensor_utils::VectorBatchVectorAssign(forget_gate_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
144 tensor_utils::VectorBatchVectorAssign(cell_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
146 tensor_utils::VectorBatchVectorAssign(output_gate_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
153 input_to_input_weights_ptr, n_cell, n_input, input_ptr_batch, n_batch, in LstmStepWithAuxInput()
[all …]
Doptional_tensor_test.cc33 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel() argument
39 n_cell_(n_cell), in LSTMOpModel()
234 const int n_cell = 4; in TEST() local
237 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, 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()
251 {n_cell, n_output}, // recurrent_to_forget_weight tensor in TEST()
252 {n_cell, n_output}, // recurrent_to_cell_weight tensor in TEST()
253 {n_cell, n_output}, // recurrent_to_output_weight tensor in TEST()
[all …]
Dunidirectional_sequence_lstm.cc102 int n_output, int n_cell) { in CheckInputTensorDimensions() argument
115 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
122 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
128 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
136 n_cell); in CheckInputTensorDimensions()
145 n_cell); in CheckInputTensorDimensions()
152 TF_LITE_ENSURE_EQ(context, recurrent_to_cell_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
169 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
176 TF_LITE_ENSURE_EQ(context, cell_to_forget_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
183 TF_LITE_ENSURE_EQ(context, cell_to_output_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
[all …]
Dlstm.cc118 int n_output, int n_cell, in CheckInputTensorDimensions() argument
133 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
140 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
146 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
154 n_cell); in CheckInputTensorDimensions()
163 n_cell); in CheckInputTensorDimensions()
170 TF_LITE_ENSURE_EQ(context, recurrent_to_cell_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
187 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
194 TF_LITE_ENSURE_EQ(context, cell_to_forget_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
201 TF_LITE_ENSURE_EQ(context, cell_to_output_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
[all …]
Dbidirectional_sequence_lstm.cc159 int n_cell, int input_to_input_weights_tensor, in CheckLstmTensorDimensionsAndTypes() argument
181 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
190 TF_LITE_ENSURE_EQ(context, input_to_input_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
199 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
207 TF_LITE_ENSURE_EQ(context, input_to_output_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
217 n_cell); in CheckLstmTensorDimensionsAndTypes()
228 n_cell); in CheckLstmTensorDimensionsAndTypes()
237 TF_LITE_ENSURE_EQ(context, recurrent_to_cell_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
256 TF_LITE_ENSURE_EQ(context, cell_to_input_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
265 TF_LITE_ENSURE_EQ(context, cell_to_forget_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
[all …]
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2259 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg, in LSTMOpModel() argument
2266 n_cell_(n_cell), in LSTMOpModel()
2571 const int n_cell = 4; in TEST_F() local
2574 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, 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()
2587 {n_cell, n_output}, // recurrent_to_input_weight_tensor in TEST_F()
2588 {n_cell, n_output}, // recurrent_to_forget_weight_tensor in TEST_F()
[all …]