• Home
  • Raw
  • Download

Lines Matching refs:output_channels

137   inline ConvHWC2SpCHWMicrokernelTester& output_channels(size_t output_channels) {  in output_channels()  function
138 assert(output_channels >= 1); in output_channels()
139 this->output_channels_ = output_channels; in output_channels()
143 inline size_t output_channels() const { in output_channels() function
148 …return output_channels() % output_channels_tile() == 0 ? output_channels() : output_channels() / o… in packed_output_channels()
242 return output_channels(); in output_pixel_stride()
304 …std::vector<float> kernel(output_channels() * kernel_height() * kernel_width() * input_channels());
305 std::vector<float> bias(output_channels());
306 std::vector<float> output(batch_size() * output_channels() * output_height() * output_width());
307 …std::vector<float> output_ref(batch_size() * output_channels() * output_height() * output_width());
318 output_channels(),
328 for (size_t oc = 0; oc < output_channels(); oc++) {
345 … output_ref[((i * output_channels() + oc) * output_height() + oy) * output_width() + ox] = acc;
379 padding_top(), output_channels(),
388 for (size_t c = 0; c < output_channels(); c++) {
389 …ASSERT_GE(output[((i * output_channels() + c) * output_height() + y) * output_width() + x], output…
391 …ASSERT_LE(output[((i * output_channels() + c) * output_height() + y) * output_width() + x], output…
394 … output_ref[((i * output_channels() + c) * output_height() + y) * output_width() + x],
395 output[((i * output_channels() + c) * output_height() + y) * output_width() + x],
396 …1.0e-4 * std::abs(output_ref[((i * output_channels() + c) * output_height() + y) * output_width() …