clDNN
roi_pooling.h
1 /*
2 // Copyright (c) 2017 Intel Corporation
3 //
4 // Licensed under the Apache License, Version 2.0 (the "License");
5 // you may not use this file except in compliance with the License.
6 // You may obtain a copy of the License at
7 //
8 // http://www.apache.org/licenses/LICENSE-2.0
9 //
10 // Unless required by applicable law or agreed to in writing, software
11 // distributed under the License is distributed on an "AS IS" BASIS,
12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 // See the License for the specific language governing permissions and
14 // limitations under the License.
15 */
16 
18 #ifndef ROI_POOLING_H
19 #define ROI_POOLING_H
20 
21 #include "cldnn.h"
28 
29 #ifdef __cplusplus
30 extern "C" {
31 #endif
32 
33 
36 int32_t mode;
37 
39 int pooled_width;
41 int pooled_height;
43 float spatial_scale;
44 
46 int group_sz;
47 CLDNN_END_PRIMITIVE_DESC(roi_pooling)
48 
49 CLDNN_DECLARE_PRIMITIVE_TYPE_ID(roi_pooling);
50 
51 #ifdef __cplusplus
52 }
53 #endif
54 
58 #endif /* ROI_POOLING_H */
59 
#define CLDNN_BEGIN_PRIMITIVE_DESC(PType)
Begin primitive description definition.
Definition: cldnn.h:398
#define CLDNN_END_PRIMITIVE_DESC(PType)
Close primitive descriptor definition.
Definition: cldnn.h:409