• Home
  • Raw
  • Download

Lines Matching refs:output_channels

37   inline FullyConnectedOperatorTester& output_channels(size_t output_channels) {  in output_channels()  function
38 assert(output_channels >= 1); in output_channels()
39 this->output_channels_ = output_channels; in output_channels()
43 inline size_t output_channels() const { in output_channels() function
80 return output_channels(); in output_stride()
82 assert(this->output_stride_ >= output_channels()); in output_stride()
140 std::vector<uint8_t> kernel(output_channels() * input_channels()); in TestQU8()
141 std::vector<int32_t> bias(output_channels()); in TestQU8()
142 std::vector<uint8_t> output((batch_size() - 1) * output_stride() + output_channels()); in TestQU8()
143 std::vector<int32_t> accumulators(batch_size() * output_channels()); in TestQU8()
144 std::vector<double> output_ref(batch_size() * output_channels()); in TestQU8()
158 for (size_t oc = 0; oc < output_channels(); oc++) { in TestQU8()
159 accumulators[i * output_channels() + oc] = bias[oc]; in TestQU8()
167 for (size_t oc = 0; oc < output_channels(); oc++) { in TestQU8()
169 accumulators[i * output_channels() + oc] += in TestQU8()
171 (int32_t(kernel[ic * output_channels() + oc]) - int32_t(kernel_zero_point)); in TestQU8()
177 for (size_t oc = 0; oc < output_channels(); oc++) { in TestQU8()
179 accumulators[i * output_channels() + oc] += in TestQU8()
208 input_channels(), output_channels(), in TestQU8()
232 for (size_t c = 0; c < output_channels(); c++) { in TestQU8()
238 output_ref[i * output_channels() + c], in TestQU8()
254 std::vector<float> kernel(output_channels() * input_channels()); in TestF32()
255 std::vector<float> bias(output_channels()); in TestF32()
256 std::vector<float> output((batch_size() - 1) * output_stride() + output_channels()); in TestF32()
257 std::vector<float> output_ref(batch_size() * output_channels()); in TestF32()
268 for (size_t oc = 0; oc < output_channels(); oc++) { in TestF32()
269 output_ref[i * output_channels() + oc] = bias[oc]; in TestF32()
277 for (size_t oc = 0; oc < output_channels(); oc++) { in TestF32()
279 output_ref[i * output_channels() + oc] += in TestF32()
280 input[i * input_stride() + ic] * kernel[ic * output_channels() + oc]; in TestF32()
286 for (size_t oc = 0; oc < output_channels(); oc++) { in TestF32()
288 output_ref[i * output_channels() + oc] += in TestF32()
315 input_channels(), output_channels(), in TestF32()
337 for (size_t c = 0; c < output_channels(); c++) { in TestF32()
343 output_ref[i * output_channels() + c], in TestF32()
345 1.0e-4 * std::abs(output_ref[i * output_channels() + c])) in TestF32()