Searched refs:weight_shape_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | conv_2d_helpers.cc | 170 weight_shape_ = {weights_height_size, weights_width_size, in InitializeWeightsNodes()
199 weight_shape_.data(), reinterpret_cast<char*>(hwcn.data()), in InitializeWeightsNodes()
207 weight_shape_ = {weights_height_size, weights_width_size, input_depth, in InitializeWeightsNodes()
218 weight_shape_.data(), reinterpret_cast<char*>(converted_data.data()), in InitializeWeightsNodes()
222 weight_shape_.data(), weights_tensor.data.raw, in InitializeWeightsNodes()
|
| D | conv_2d_builder.h | 65 std::vector<int> weight_shape_; variable
|
| D | conv_2d_builder.cc | 236 input_height_size, input_width_size, weight_shape_[0], weight_shape_[1], in PopulateSubGraph()
|