Searched refs:output_h_shape (Results 1 – 1 of 1) sorted by relevance
97 auto output_h_shape = input_h_shape; in __anon815c5c9f0302() local99 (rnn_mode == "lstm") ? output_h_shape : c->MakeShape({}); in __anon815c5c9f0302()101 c->set_output(1, output_h_shape); in __anon815c5c9f0302()140 auto output_h_shape = input_h_shape; in __anon815c5c9f0402() local142 (rnn_mode == "lstm") ? output_h_shape : c->MakeShape({}); in __anon815c5c9f0402()144 c->set_output(1, output_h_shape); in __anon815c5c9f0402()189 auto output_h_shape = input_h_shape; in __anon815c5c9f0502() local193 c->set_output(1, output_h_shape); in __anon815c5c9f0502()