Searched refs:weights_dim2 (Results 1 - 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| H A D | identify_lstm_merge_inputs.cc | 77 int weights_dim2 = num_input + num_output; local
78 Shape shape = Shape({weights_dim1, weights_dim2});
81 buffer.data.resize(weights_dim1 * weights_dim2);
85 buffer, weights_dim2,
88 buffer, weights_dim2,
91 buffer, weights_dim2,
95 buffer, weights_dim2,
99 buffer, weights_dim2,
103 buffer, weights_dim2,
107 buffer, weights_dim2,
[all...] |
Completed in 265 milliseconds