1 /* 2 * Copyright (c) Facebook, Inc. and its affiliates. 3 * All rights reserved. 4 * 5 * This source code is licensed under the BSD-style license found in the 6 * LICENSE file in the root directory of this source tree. 7 */ 8 9 #pragma once 10 11 #include <stddef.h> 12 #include <stdint.h> 13 14 #include<pytorch_qnnpack.h> 15 #include <qnnpack/common.h> 16 17 #ifdef __cplusplus 18 extern "C" { 19 #endif 20 21 PYTORCH_QNNP_INTERNAL void pytorch_qnnp_indirection_init_conv3d( 22 pytorch_qnnp_operator_t op, 23 size_t output_tile_size, 24 size_t tiled_output_size); 25 26 PYTORCH_QNNP_INTERNAL void pytorch_qnnp_indirection_init_dwconv( 27 pytorch_qnnp_operator_t op, 28 size_t batch_start); 29 30 PYTORCH_QNNP_INTERNAL void pytorch_qnnp_indirection_init_deconv2d( 31 pytorch_qnnp_operator_t op, 32 size_t output_tile_size, 33 size_t tiled_output_size); 34 35 PYTORCH_QNNP_INTERNAL void pytorch_qnnp_indirection_init_maxpool2d( 36 pytorch_qnnp_operator_t op, 37 size_t batch_start); 38 39 PYTORCH_QNNP_INTERNAL void pytorch_qnnp_indirection_set_step_dimensions( 40 pytorch_qnnp_operator_t op); 41 42 #ifdef __cplusplus 43 } /* extern "C" */ 44 #endif 45