Lines Matching refs:output_channels
45 inline FullyConnectedOperatorTester& output_channels(size_t output_channels) { in output_channels() function
46 assert(output_channels >= 1); in output_channels()
47 this->output_channels_ = output_channels; in output_channels()
51 inline size_t output_channels() const { in output_channels() function
88 return output_channels(); in output_stride()
90 assert(this->output_stride_ >= output_channels()); in output_stride()
171 std::vector<int8_t> kernel(output_channels() * input_channels()); in TestQS8()
172 std::vector<int32_t> bias(output_channels()); in TestQS8()
173 std::vector<int8_t> output((batch_size() - 1) * output_stride() + output_channels()); in TestQS8()
174 std::vector<int32_t> accumulators(batch_size() * output_channels()); in TestQS8()
175 std::vector<double> output_ref(batch_size() * output_channels()); in TestQS8()
188 for (size_t oc = 0; oc < output_channels(); oc++) { in TestQS8()
189 accumulators[i * output_channels() + oc] = bias[oc]; in TestQS8()
197 for (size_t oc = 0; oc < output_channels(); oc++) { in TestQS8()
199 accumulators[i * output_channels() + oc] += in TestQS8()
201 int32_t(kernel[ic * output_channels() + oc]); in TestQS8()
207 for (size_t oc = 0; oc < output_channels(); oc++) { in TestQS8()
209 accumulators[i * output_channels() + oc] += in TestQS8()
247 input_channels(), output_channels(), in TestQS8()
289 input_channels(), output_channels(), input_stride(), in TestQS8()
327 for (size_t c = 0; c < output_channels(); c++) { in VerifyQS8()
332 ASSERT_NEAR(output_ref[i * output_channels() + c], in VerifyQS8()
351 std::vector<uint8_t> kernel(output_channels() * input_channels()); in TestQU8()
352 std::vector<int32_t> bias(output_channels()); in TestQU8()
353 std::vector<uint8_t> output((batch_size() - 1) * output_stride() + output_channels()); in TestQU8()
354 std::vector<int32_t> accumulators(batch_size() * output_channels()); in TestQU8()
355 std::vector<double> output_ref(batch_size() * output_channels()); in TestQU8()
369 for (size_t oc = 0; oc < output_channels(); oc++) { in TestQU8()
370 accumulators[i * output_channels() + oc] = bias[oc]; in TestQU8()
378 for (size_t oc = 0; oc < output_channels(); oc++) { in TestQU8()
380 accumulators[i * output_channels() + oc] += in TestQU8()
382 (int32_t(kernel[ic * output_channels() + oc]) - int32_t(kernel_zero_point)); in TestQU8()
388 for (size_t oc = 0; oc < output_channels(); oc++) { in TestQU8()
390 accumulators[i * output_channels() + oc] += in TestQU8()
428 input_channels(), output_channels(), in TestQU8()
469 input_channels(), output_channels(), input_stride(), in TestQU8()
505 for (size_t c = 0; c < output_channels(); c++) { in VerifyQU8()
510 ASSERT_NEAR(output_ref[i * output_channels() + c], in VerifyQU8()
527 std::vector<float> kernel(output_channels() * input_channels()); in TestF32()
528 std::vector<float> bias(output_channels()); in TestF32()
529 std::vector<float> output((batch_size() - 1) * output_stride() + output_channels()); in TestF32()
530 std::vector<float> output_ref(batch_size() * output_channels()); in TestF32()
541 for (size_t oc = 0; oc < output_channels(); oc++) { in TestF32()
542 output_ref[i * output_channels() + oc] = bias[oc]; in TestF32()
550 for (size_t oc = 0; oc < output_channels(); oc++) { in TestF32()
552 output_ref[i * output_channels() + oc] += in TestF32()
553 input[i * input_stride() + ic] * kernel[ic * output_channels() + oc]; in TestF32()
559 for (size_t oc = 0; oc < output_channels(); oc++) { in TestF32()
561 output_ref[i * output_channels() + oc] += in TestF32()
597 input_channels(), output_channels(), in TestF32()
635 input_channels(), output_channels(), input_stride(), in TestF32()
669 for (size_t c = 0; c < output_channels(); c++) { in VerifyF32()
674 ASSERT_NEAR(output_ref[i * output_channels() + c], in VerifyF32()
676 1.0e-4 * std::abs(output_ref[i * output_channels() + c])) in VerifyF32()
698 std::vector<uint16_t> kernel(output_channels() * input_channels()); in TestF16()
700 std::vector<uint16_t> bias(output_channels()); in TestF16()
702 std::vector<uint16_t> output((batch_size() - 1) * output_stride() + output_channels()); in TestF16()
703 std::vector<float> output_ref(batch_size() * output_channels()); in TestF16()
716 for (size_t oc = 0; oc < output_channels(); oc++) { in TestF16()
717 output_ref[i * output_channels() + oc] = fp16_ieee_to_fp32_value(bias[oc]); in TestF16()
725 for (size_t oc = 0; oc < output_channels(); oc++) { in TestF16()
727 output_ref[i * output_channels() + oc] += in TestF16()
728 …lue(input[i * input_stride() + ic]) * fp16_ieee_to_fp32_value(kernel[ic * output_channels() + oc]); in TestF16()
734 for (size_t oc = 0; oc < output_channels(); oc++) { in TestF16()
736 output_ref[i * output_channels() + oc] += in TestF16()
785 input_channels(), output_channels(), in TestF16()
823 input_channels(), output_channels(), input_stride(), in TestF16()
860 for (size_t c = 0; c < output_channels(); c++) { in VerifyF16()
866 output_ref[i * output_channels() + c], in VerifyF16()
868 1.0e-2f * std::abs(output_ref[i * output_channels() + c])) in VerifyF16()