Searched refs:weights_width_size (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | conv_2d_helpers.cc | 150 int weights_batch_size, weights_height_size, weights_width_size, in InitializeWeightsNodes() local
154 GetDims(&weights_batch_size, &weights_height_size, &weights_width_size, in InitializeWeightsNodes()
170 weight_shape_ = {weights_height_size, weights_width_size, in InitializeWeightsNodes()
173 weights_width_size, weights_depth_size}); in InitializeWeightsNodes()
174 RuntimeShape hwcn_shape({weights_height_size, weights_width_size, in InitializeWeightsNodes()
207 weight_shape_ = {weights_height_size, weights_width_size, input_depth, in InitializeWeightsNodes()
|