1 // Copyright (c) Facebook, Inc. and its affiliates. 2 // All rights reserved. 3 // 4 // Copyright 2019 Google LLC 5 // 6 // This source code is licensed under the BSD-style license found in the 7 // LICENSE file in the root directory of this source tree. 8 9 #pragma once 10 11 #include <stddef.h> 12 #include <stdint.h> 13 #include <stdbool.h> 14 15 #include <xnnpack.h> 16 #include <xnnpack/common.h> 17 18 #ifdef __cplusplus 19 extern "C" { 20 #endif 21 22 XNN_INTERNAL void xnn_indirection_init_conv2d( 23 xnn_operator_t op, 24 size_t output_tile_size, 25 uint32_t log2_element_size); 26 27 XNN_INTERNAL void xnn_indirection_init_dwconv2d( 28 xnn_operator_t op, 29 size_t batch_start, 30 size_t step_height, 31 size_t step_width, 32 uint32_t log2_element_size); 33 34 XNN_INTERNAL void xnn_indirection_init_deconv2d( 35 xnn_operator_t op, 36 size_t output_tile_size, 37 uint32_t log2_element_size); 38 39 XNN_INTERNAL void xnn_indirection_init_subconv2d( 40 xnn_operator_t op, 41 size_t output_tile_size, 42 uint32_t log2_element_size); 43 44 XNN_INTERNAL void xnn_indirection_init_maxpool2d( 45 xnn_operator_t op, 46 size_t step_height, 47 size_t step_width, 48 uint32_t log2_element_size); 49 50 XNN_INTERNAL void xnn_indirection_init_resize_bilinear2d_f32( 51 size_t input_pixel_stride, 52 size_t input_height, 53 size_t input_width, 54 size_t output_height, 55 size_t output_width, 56 const void* input, 57 const void** indirection_buffer, 58 float* packed_weights, 59 bool align_corners, 60 bool tensorflow_legacy); 61 62 XNN_INTERNAL void xnn_indirection_init_unpool2d( 63 xnn_operator_t op, 64 size_t batch_start, 65 uint32_t log2_element_size); 66 67 #ifdef __cplusplus 68 } // extern "C" 69 #endif 70