inference-engine/thirdparty/clDNN/api/C/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
  17 ///////////////////////////////////////////////////////////////////////////////////////////////////
  18 #ifndef ROI_POOLING_H
  19 #define ROI_POOLING_H
  20
  21 #include <stdbool.h>
  22 #include "cldnn.h"
  23 /// @addtogroup c_api C API
  24 /// @{
  25 /// @addtogroup c_topology Network Topology
  26 /// @{
  27 /// @addtogroup c_primitives Primitives
  28 /// @{
  29
  30 #ifdef __cplusplus
  31 extern "C" {
  32 #endif
  33
  34
  35 CLDNN_BEGIN_PRIMITIVE_DESC(roi_pooling)
  36 /// @brief Pooling method. See #cldnn_pooling_mode.
  37 int32_t mode;
  38 /// @brief True, if pooling is position sensitive (PSROIPoolng)
  39 bool position_sensitive;
  40 /// @brief Output width.
  41 int pooled_width;
  42 /// @brief Output height.
  43 int pooled_height;
  44 /// @brief Count of sub bins in x spatial dimension
  45 int spatial_bins_x;
  46 /// @brief Count of sub bins in y spatial dimension
  47 int spatial_bins_y;
  48 /// @brief Output features count (applied for position sensitive case only)
  49 int output_dim;
  50 /// @brief Ratio of the coordinates used in RoIs to the width (and height) of the input data.
  51 float spatial_scale;
  52 CLDNN_END_PRIMITIVE_DESC(roi_pooling)
  53
  54 CLDNN_DECLARE_PRIMITIVE_TYPE_ID(roi_pooling);
  55
  56 #ifdef __cplusplus
  57 }
  58 #endif
  59
  60 /// @}
  61 /// @}
  62 /// @}
  63 #endif /* ROI_POOLING_H */
  64