Searched refs:input_c_shape (Results 1 – 2 of 2) sorted by relevance
176 auto input_c_shape = c->input(2); in __anon815c5c9f0502() local191 (rnn_mode == "lstm") ? input_c_shape : c->MakeShape({}); in __anon815c5c9f0502()227 auto input_c_shape = c->input(2); in __anon815c5c9f0602() local231 c->set_output(2, input_c_shape); in __anon815c5c9f0602()264 auto input_c_shape = c->input(2); in __anon815c5c9f0702() local268 c->set_output(2, input_c_shape); in __anon815c5c9f0702()304 auto input_c_shape = c->input(2); in __anon815c5c9f0802() local308 c->set_output(2, input_c_shape); in __anon815c5c9f0802()
114 std::vector<int> input_c_shape = {num_layers * dir_count, batch_size, in TEST() local124 shape_to_str(input_c_shape), ";", "[?]", ";", in TEST()