Lines Matching refs:output_channels

137   inline ConvHWCMicrokernelTester& 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()
298 …std::vector<float> kernel(output_channels() * kernel_height() * kernel_width() * input_channels());
299     std::vector<float> bias(output_channels());
300 …tch_size() * ((output_height() * output_width() - 1) * output_pixel_stride() + output_channels()));
301 …std::vector<float> output_ref(batch_size() * output_height() * output_width() * output_channels());
312         output_channels(),
322             for (size_t oc = 0; oc < output_channels(); oc++) {
339 …     output_ref[((i * output_height() + oy) * output_width() + ox) * output_channels() + oc] = acc;
373         padding_top(), output_channels(),
382             for (size_t c = 0; c < output_channels(); c++) {
388 …              output_ref[((i * output_height() + y) * output_width() + x) * output_channels() + c],
390 …4 * std::abs(output_ref[((i * output_height() + y) * output_width() + x) * output_channels() + c]))