Searched refs:input_h_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | cudnn_rnn_ops.cc | 85 auto input_h_shape = c->input(1); in __anon815c5c9f0302() local
88 auto num_units = c->Dim(input_h_shape, 2); in __anon815c5c9f0302()
97 auto output_h_shape = input_h_shape; in __anon815c5c9f0302()
128 auto input_h_shape = c->input(1); in __anon815c5c9f0402() local
131 auto num_units = c->Dim(input_h_shape, 2); in __anon815c5c9f0402()
140 auto output_h_shape = input_h_shape; in __anon815c5c9f0402()
175 auto input_h_shape = c->input(1); in __anon815c5c9f0502() local
179 auto num_units = c->Dim(input_h_shape, 2); in __anon815c5c9f0502()
189 auto output_h_shape = input_h_shape; in __anon815c5c9f0502()
226 auto input_h_shape = c->input(1); in __anon815c5c9f0602() local
[all …]
|
| D | cudnn_rnn_ops_test.cc | 48 std::vector<int> input_h_shape = {num_layers * dir_count, batch_size, in TEST() local
56 shape_to_str(input_shape), ";", shape_to_str(input_h_shape), ";", in TEST()
57 shape_to_str(input_h_shape), ";", "[?]"); in TEST()
80 std::vector<int> input_h_shape = {num_layers * dir_count, batch_size, in TEST() local
88 shape_to_str(input_shape), ";", shape_to_str(input_h_shape), ";", in TEST()
89 shape_to_str(input_h_shape), ";", "[?]"); in TEST()
112 std::vector<int> input_h_shape = {num_layers * dir_count, batch_size, in TEST() local
123 shape_to_str(input_shape), ";", shape_to_str(input_h_shape), ";", in TEST()
|