Lines Matching +full:- +full:oc

3 // This source code is licensed under the BSD-style license found in the
27 this->input_height_ = input_height; in input_size()
28 this->input_width_ = input_width; in input_size()
34 this->input_height_ = input_height; in input_height()
39 return this->input_height_; in input_height()
44 this->input_width_ = input_width; in input_width()
49 return this->input_width_; in input_width()
62 this->block_size_ = block_size; in block_size()
67 return this->block_size_; in block_size()
76 this->output_channels_ = output_channels; in output_channels()
81 return this->output_channels_; in output_channels()
86 this->batch_size_ = batch_size; in batch_size()
91 return this->batch_size_; in batch_size()
96 this->input_channels_stride_ = input_channels_stride; in input_channels_stride()
101 if (this->input_channels_stride_ == 0) { in input_channels_stride()
104 assert(this->input_channels_stride_ >= input_channels()); in input_channels_stride()
105 return this->input_channels_stride_; in input_channels_stride()
111 this->output_channels_stride_ = output_channels_stride; in output_channels_stride()
116 if (this->output_channels_stride_ == 0) { in output_channels_stride()
119 assert(this->output_channels_stride_ >= output_channels()); in output_channels_stride()
120 return this->output_channels_stride_; in output_channels_stride()
125 this->iterations_ = iterations; in iterations()
130 return this->iterations_; in iterations()
141 … (batch_size() * input_height() * input_width() - 1) * input_channels_stride() + input_channels()); in TestNHWCxX8()
143 …(batch_size() * output_height() * output_width() - 1) * output_channels_stride() + output_channels… in TestNHWCxX8()
176 for (size_t oc = 0; oc < output_channels(); oc++) { in TestNHWCxX8() local
179 (by * block_size() + bx) * output_channels() + oc; in TestNHWCxX8()
182 output_channels_stride() + oc; in TestNHWCxX8()
189 << ", output channel: " << oc << " / " << output_channels() in TestNHWCxX8()
207 … (batch_size() * input_height() * input_width() - 1) * input_channels_stride() + input_channels()); in TestNHWCxX16()
209 …(batch_size() * output_height() * output_width() - 1) * output_channels_stride() + output_channels… in TestNHWCxX16()
242 for (size_t oc = 0; oc < output_channels(); oc++) { in TestNHWCxX16() local
245 (by * block_size() + bx) * output_channels() + oc; in TestNHWCxX16()
248 output_channels_stride() + oc; in TestNHWCxX16()
255 << ", output channel: " << oc << " / " << output_channels() in TestNHWCxX16()
273 … (batch_size() * input_height() * input_width() - 1) * input_channels_stride() + input_channels()); in TestNHWCxX32()
275 …(batch_size() * output_height() * output_width() - 1) * output_channels_stride() + output_channels… in TestNHWCxX32()
308 for (size_t oc = 0; oc < output_channels(); oc++) { in TestNHWCxX32() local
311 (by * block_size() + bx) * output_channels() + oc; in TestNHWCxX32()
314 output_channels_stride() + oc; in TestNHWCxX32()
321 << ", output channel: " << oc << " / " << output_channels() in TestNHWCxX32()
339 …((batch_size() - 1) * input_channels_stride() + input_channels()) * input_height() * input_width()… in TestNCHW2NHWCxX32()
341 …(batch_size() * output_height() * output_width() - 1) * output_channels_stride() + output_channels… in TestNCHW2NHWCxX32()
374 for (size_t oc = 0; oc < output_channels(); oc++) { in TestNCHW2NHWCxX32() local
377 … (((by * block_size() + bx) * output_channels() + oc) * input_height() + iy) * input_width() + ix; in TestNCHW2NHWCxX32()
380 output_channels_stride() + oc; in TestNCHW2NHWCxX32()
387 << ", output channel: " << oc << " / " << output_channels() in TestNCHW2NHWCxX32()