Lines Matching refs:output_channels
71 return output_channels() * block_size() * block_size(); in input_channels()
74 inline DepthToSpaceOperatorTester& output_channels(size_t output_channels) { in output_channels() function
75 assert(output_channels != 0); in output_channels()
76 this->output_channels_ = output_channels; in output_channels()
80 inline size_t output_channels() const { in output_channels() function
117 return output_channels(); in output_channels_stride()
119 assert(this->output_channels_stride_ >= output_channels()); in output_channels_stride()
141 …tch_size() * output_height() * output_width() - 1) * output_channels_stride() + output_channels()); in TestNHWCxX32()
152 output_channels(), input_channels_stride(), output_channels_stride(), in TestNHWCxX32()
174 for (size_t oc = 0; oc < output_channels(); oc++) { in TestNHWCxX32()
177 (by * block_size() + bx) * output_channels() + oc; in TestNHWCxX32()
187 << ", output channel: " << oc << " / " << output_channels() in TestNHWCxX32()
207 …tch_size() * output_height() * output_width() - 1) * output_channels_stride() + output_channels()); in TestNCHW2NHWCxX32()
218 output_channels(), input_channels_stride(), output_channels_stride(), in TestNCHW2NHWCxX32()
240 for (size_t oc = 0; oc < output_channels(); oc++) { in TestNCHW2NHWCxX32()
243 … (((by * block_size() + bx) * output_channels() + oc) * input_height() + iy) * input_width() + ix; in TestNCHW2NHWCxX32()
253 << ", output channel: " << oc << " / " << output_channels() in TestNCHW2NHWCxX32()