Home
last modified time | relevance | path

Searched refs:weights_shape_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dmatmul_builder.h41 std::vector<int> weights_shape_, bias_shape_; variable
68 std::vector<int> weights_shape_, bias_shape_; variable
Dmatmul_builder.cc171 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()