Lines Matching refs:output_pixel_stride
247 inline ArgmaxPoolingOperatorTester& output_pixel_stride(size_t output_pixel_stride) { in output_pixel_stride() argument
248 assert(output_pixel_stride != 0); in output_pixel_stride()
249 this->output_pixel_stride_ = output_pixel_stride; in output_pixel_stride()
253 inline size_t output_pixel_stride() const { in output_pixel_stride() function
337 …at> output((batch_size() * output_height() * output_width() - 1) * output_pixel_stride() + channel… in TestF32()
384 channels(), input_pixel_stride(), output_pixel_stride(), in TestF32()
408 … output[((i * output_height() + y) * output_width() + x) * output_pixel_stride() + c]) << in TestF32()
429 (batch_size() * output_height() * output_width() - 1) * output_pixel_stride() + channels(), in TestSetupF32()
430 …(next_batch_size() * next_output_height() * next_output_width() - 1) * output_pixel_stride() + cha… in TestSetupF32()
480 channels(), input_pixel_stride(), output_pixel_stride(), in TestSetupF32()
501 … output[((i * output_height() + y) * output_width() + x) * output_pixel_stride() + c]) in TestSetupF32()
568 … output[((i * next_output_height() + y) * next_output_width() + x) * output_pixel_stride() + c]) in TestSetupF32()
572 … index[((i * next_output_height() + y) * next_output_width() + x) * output_pixel_stride() + c]) in TestSetupF32()