Searched refs:use_cifg (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Doptional_tensor_test.cc33 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 Dlstm.cc161 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 Dunidirectional_sequence_lstm.cc161 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 Dlstm_test.cc35 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 Dunidirectional_sequence_lstm_test.cc36 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