Lines Matching refs:output_channels
42 const size_t output_channels = state.range(12); in xnnpack_deconvolution_qu8() local
62 std::vector<uint8_t> kernel(output_channels * kernel_height * kernel_width * input_channels); in xnnpack_deconvolution_qu8()
64 std::vector<int32_t> bias(output_channels); in xnnpack_deconvolution_qu8()
66 const size_t output_elements = batch_size * output_height * output_width * output_channels; in xnnpack_deconvolution_qu8()
86 /*groups=*/1, input_channels, output_channels, in xnnpack_deconvolution_qu8()
87 /*input_pixel_stride=*/input_channels, /*output_pixel_stride=*/output_channels, in xnnpack_deconvolution_qu8()
144 input_channels * output_channels * in xnnpack_deconvolution_qu8()
163 const size_t output_channels = state.range(12); in xnnpack_deconvolution_f32() local
181 std::vector<float> kernel(output_channels * kernel_height * kernel_width * input_channels); in xnnpack_deconvolution_f32()
183 std::vector<float> bias(output_channels); in xnnpack_deconvolution_f32()
185 const size_t output_elements = batch_size * output_height * output_width * output_channels; in xnnpack_deconvolution_f32()
205 /*groups=*/1, input_channels, output_channels, in xnnpack_deconvolution_f32()
206 /*input_pixel_stride=*/input_channels, /*output_pixel_stride=*/output_channels, in xnnpack_deconvolution_f32()
262 input_channels * output_channels * in xnnpack_deconvolution_f32()
281 const size_t output_channels = state.range(12); in tflite_deconvolution_f32() local
305 std::vector<float> kernel(output_channels * kernel_height * kernel_width * input_channels); in tflite_deconvolution_f32()
327 static_cast<int32_t>(output_channels) in tflite_deconvolution_f32()
330 static_cast<int32_t>(output_channels), in tflite_deconvolution_f32()
441 input_channels * output_channels * in tflite_deconvolution_f32()