• Home
  • Raw
  • Download

Lines Matching refs:block_size

53     return input_height() * block_size();  in output_height()
57 return input_width() * block_size(); in output_width()
60 inline DepthToSpaceOperatorTester& block_size(size_t block_size) { in block_size() function
61 assert(block_size >= 2); in block_size()
62 this->block_size_ = block_size; in block_size()
66 inline size_t block_size() const { in block_size() function
71 return output_channels() * block_size() * block_size(); in input_channels()
153 block_size(), 0, &depth_to_space_op)); in TestNHWCxX32()
171 for (size_t by = 0; by < block_size(); by++) { in TestNHWCxX32()
173 for (size_t bx = 0; bx < block_size(); bx++) { in TestNHWCxX32()
177 (by * block_size() + bx) * output_channels() + oc; in TestNHWCxX32()
179 … ((i * output_height() + iy * block_size() + by) * output_width() + ix * block_size() + bx) * in TestNHWCxX32()
185 << ", block x: " << bx << " / " << block_size() in TestNHWCxX32()
186 << ", block y: " << by << " / " << block_size() in TestNHWCxX32()
219 block_size(), 0, &depth_to_space_op)); in TestNCHW2NHWCxX32()
237 for (size_t by = 0; by < block_size(); by++) { in TestNCHW2NHWCxX32()
239 for (size_t bx = 0; bx < block_size(); bx++) { in TestNCHW2NHWCxX32()
243 … (((by * block_size() + bx) * output_channels() + oc) * input_height() + iy) * input_width() + ix; in TestNCHW2NHWCxX32()
245 … ((i * output_height() + iy * block_size() + by) * output_width() + ix * block_size() + bx) * in TestNCHW2NHWCxX32()
251 << ", block x: " << bx << " / " << block_size() in TestNCHW2NHWCxX32()
252 << ", block y: " << by << " / " << block_size() in TestNCHW2NHWCxX32()