| // Copyright (c) Facebook, Inc. and its affiliates. |
| // All rights reserved. |
| // |
| // Copyright 2019 Google LLC |
| // |
| // This source code is licensed under the BSD-style license found in the |
| // LICENSE file in the root directory of this source tree. |
| |
| #pragma once |
| |
| #include <stddef.h> |
| #include <stdint.h> |
| |
| #include <xnnpack.h> |
| #include <xnnpack/common.h> |
| |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| XNN_INTERNAL void xnn_indirection_init_conv2d( |
| xnn_operator_t op, |
| size_t output_tile_size, |
| uint32_t log2_element_size); |
| |
| XNN_INTERNAL void xnn_indirection_init_dwconv2d( |
| xnn_operator_t op, |
| size_t batch_start, |
| size_t step_height, |
| size_t step_width, |
| uint32_t log2_element_size); |
| |
| XNN_INTERNAL void xnn_indirection_init_deconv2d( |
| xnn_operator_t op, |
| size_t output_tile_size, |
| uint32_t log2_element_size); |
| |
| XNN_INTERNAL void xnn_indirection_init_subconv2d( |
| xnn_operator_t op, |
| size_t output_tile_size, |
| uint32_t log2_element_size); |
| |
| XNN_INTERNAL void xnn_indirection_init_maxpool2d( |
| xnn_operator_t op, |
| size_t batch_start, |
| size_t step_height, |
| size_t step_width, |
| uint32_t log2_element_size); |
| |
| XNN_INTERNAL void xnn_indirection_init_unpool2d( |
| xnn_operator_t op, |
| size_t batch_start, |
| uint32_t log2_element_size); |
| |
| #ifdef __cplusplus |
| } /* extern "C" */ |
| #endif |