Searched refs:use_cifg (Results 1 - 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/contrib/lite/kernels/ |
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 43 if (use_cifg) { 53 if (use_cifg) { 64 if (use_cifg) { 77 if (use_cifg) { 254 /*use_cifg=*/true, /*use_peephole=*/true,
|
H A D | lstm.cc | 161 const bool use_cifg = (input_to_input_weights == nullptr); local 163 ((cell_to_input_weights != nullptr || use_cifg) && 174 if (use_cifg) { 286 const bool use_cifg = (input_to_input_weights == nullptr); local 287 if (use_cifg) { 360 const bool use_cifg = (input_to_input_weights == nullptr); local 369 if (use_cifg) { 381 if (!use_cifg) { 393 if (!use_cifg) { 409 if (!use_cifg) { [all...] |
H A D | unidirectional_sequence_lstm.cc | 161 const bool use_cifg = (input_to_input_weights == nullptr); local 163 ((cell_to_input_weights != nullptr || use_cifg) && 174 if (use_cifg) { 288 const bool use_cifg = (input_to_input_weights == nullptr); local 289 if (use_cifg) { 363 const bool use_cifg = (input_to_input_weights == nullptr); local 372 if (use_cifg) { 386 if (!use_cifg) { 398 if (!use_cifg) { 414 if (!use_cifg) { [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 45 if (use_cifg) { 55 if (use_cifg) { 66 if (use_cifg) { 79 if (use_cifg) { 252 /*use_cifg=*/false, /*use_peephole=*/false, 362 /*use_cifg=*/true, /*use_peephole=*/true, 466 /*use_cifg=*/false, /*use_peephole=*/true,
|
H A D | unidirectional_sequence_lstm_test.cc | 36 int sequence_length, bool use_cifg, 48 if (use_cifg) { 58 if (use_cifg) { 69 if (use_cifg) { 82 if (use_cifg) { 259 n_batch, n_input, n_cell, n_output, sequence_length, /*use_cifg=*/false, 367 n_batch, n_input, n_cell, n_output, sequence_length, /*use_cifg=*/true, 468 n_batch, n_input, n_cell, n_output, sequence_length, /*use_cifg=*/false, 35 UnidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, int sequence_length, bool use_cifg, bool use_peephole, bool use_projection_weights, bool use_projection_bias, float cell_clip, float proj_clip, const std::vector<std::vector<int>>& input_shapes) argument
|
Completed in 114 milliseconds