| /external/tensorflow/tensorflow/lite/kernels/ |
| lstm_eval.cc | 123 const bool use_cifg = (input_to_input_weights_ptr == nullptr); local
131 if (!use_cifg) {
138 if (!use_cifg) {
151 if (!use_cifg) {
169 if (!use_cifg) {
188 if (!use_cifg) {
206 if (!use_cifg) {
259 if (use_cifg) {
477 const bool use_cifg = (input_to_input_weights_ptr == nullptr); local
484 if (!use_cifg) {
917 const bool use_cifg = (input_to_input_weights == nullptr); local
1120 const bool use_cifg = (input_to_input_weights == nullptr); local
[all...] |
| optional_tensor_test.cc | 33 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg,
43 if (use_cifg) {
53 if (use_cifg) {
64 if (use_cifg) {
77 if (use_cifg) {
238 /*use_cifg=*/true, /*use_peephole=*/true,
|
| lstm_test.cc | 37 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg,
49 if (use_cifg) {
59 if (use_cifg) {
70 if (use_cifg) {
83 if (use_cifg) {
112 if (use_cifg) {
272 bool use_cifg, bool use_peephole,
277 : LSTMOpModel(n_batch, n_input, n_cell, n_output, use_cifg, use_peephole,
457 /*use_cifg=*/false, /*use_peephole=*/false,
515 /*use_cifg=*/false, /*use_peephole=*/false
[all...] |
| unidirectional_sequence_lstm.cc | 187 const bool use_cifg = (input_to_input_weights == nullptr); local
189 ((cell_to_input_weights != nullptr || use_cifg) &&
200 if (use_cifg) {
328 const bool use_cifg = (input_to_input_weights == nullptr); local
331 if (use_cifg) {
|
| lstm.cc | 130 const bool use_cifg = (input_to_input_weights == nullptr); local
131 if (!use_cifg) {
206 ((cell_to_input_weights != nullptr || use_cifg) &&
217 if (use_cifg) {
265 if (use_cifg) {
404 const bool use_cifg = (input_to_input_weights == nullptr); local
407 if (use_cifg) {
[all...] |
| unidirectional_sequence_lstm_test.cc | 35 int sequence_length, bool time_major, bool use_cifg,
48 if (use_cifg) {
58 if (use_cifg) {
69 if (use_cifg) {
82 if (use_cifg) {
244 bool time_major, bool use_cifg, bool use_peephole,
250 use_cifg, use_peephole, use_projection_weights, use_projection_bias,
460 /*time_major=*/true, /*use_cifg=*/false, /*use_peephole=*/false,
522 /*time_major=*/true, /*use_cifg=*/false, /*use_peephole=*/false,
589 /*time_major=*/true, /*use_cifg=*/false, /*use_peephole=*/false
[all...] |
| bidirectional_sequence_lstm_test.cc | 38 int sequence_length, bool use_cifg,
56 if (use_cifg) {
66 if (use_cifg) {
77 if (use_cifg) {
90 if (use_cifg) {
111 if (use_cifg) {
121 if (use_cifg) {
132 if (use_cifg) {
145 if (use_cifg) {
415 n_batch, n_input, n_cell, n_output, sequence_length, /*use_cifg=*/false
[all...] |
| bidirectional_sequence_lstm.cc | 280 const bool use_cifg = (input_to_input_weights == nullptr); local
282 ((cell_to_input_weights != nullptr || use_cifg) &&
293 if (use_cifg) {
[all...] |
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| nnapi_delegate_test.cc | [all...] |
