• Home
  • Raw
  • Download

Lines Matching refs:output_elements

77   const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride;  in xnnpack_convolution_qu8()  local
87 …zeof(uint8_t) * kernel.size() + sizeof(int32_t) * bias.size() + sizeof(uint8_t) * output_elements); in xnnpack_convolution_qu8()
88 std::vector<uint8_t> output(output_elements * num_buffers); in xnnpack_convolution_qu8()
114 input.data(), output.data() + i * output_elements, in xnnpack_convolution_qu8()
198 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_qs8() local
208 …sizeof(int8_t) * kernel.size() + sizeof(int32_t) * bias.size() + sizeof(int8_t) * output_elements); in xnnpack_convolution_qs8()
209 std::vector<int8_t> output(output_elements * num_buffers); in xnnpack_convolution_qs8()
234 input.data(), output.data() + i * output_elements, in xnnpack_convolution_qs8()
320 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_f16() local
330 sizeof(uint16_t) * (kernel.size() + bias.size() + output_elements)); in xnnpack_convolution_f16()
331 std::vector<uint16_t> output(output_elements * num_buffers); in xnnpack_convolution_f16()
355 input.data(), output.data() + i * output_elements, in xnnpack_convolution_f16()
435 const size_t output_elements = batch_size * output_height * output_width * output_pixel_stride; in xnnpack_convolution_f32() local
445 sizeof(float) * (kernel.size() + bias.size() + output_elements)); in xnnpack_convolution_f32()
446 std::vector<float> output(output_elements * num_buffers); in xnnpack_convolution_f32()
470 input.data(), output.data() + i * output_elements, in xnnpack_convolution_f32()