Searched refs:weights_shape_ (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
D | matmul_builder.h | 41 std::vector<int> weights_shape_, bias_shape_; variable 68 std::vector<int> weights_shape_, bias_shape_; variable
|
D | matmul_builder.cc | 171 weights_shape_ = {batch_size, height_size, depth_size, width_size}; in PopulateSubGraph() 196 weights_shape_.data(), reinterpret_cast<char*>(nhcw.data()), in PopulateSubGraph() 231 weights_shape_ = {batch_size, height_size, depth_size, width_size}; in PopulateSubGraph() 251 auto transposed_weights = AddOutput(sizeof(uint8_t), 4, weights_shape_); in PopulateSubGraph()
|