add pooling, transpose convolution to layers.py and update example.py

user_ops/CMakeLists.txt

@@ -10,7 +10,7 @@ project( FlexConv )
 
 list(APPEND CMAKE_MODULE_PATH ${PROJECT_SOURCE_DIR})
 
-find_package(CUDA 9 REQUIRED)
+find_package(CUDA REQUIRED)
 find_package(TensorFlow REQUIRED)
 
 

user_ops/FindTensorFlow.cmake

@@ -122,7 +122,7 @@ else()
     # However, only TensorFlow versions 1.9, 1.10 support all header files
     # for custom ops.
     set(_TensorFlow_KNOWN_VERSIONS ${TensorFlow_ADDITIONAL_VERSIONS}
-        "1.9" "1.9.0" "1.10" "1.10.0")
+        "1.9" "1.9.0" "1.10" "1.10.0", "1.11", "1.11.0")
     set(_TensorFlow_TEST_VERSIONS)
 
     if(TF_FIND_VERSION)
@@ -270,4 +270,4 @@ SET(TensorFlow_INCLUDE_DIR ${TensorFlow_INCLUDE_DIR} CACHE PATH "path to tensorf
 SET(TensorFlow_VERSION ${TensorFlow_VERSION} CACHE INTERNAL "The Python executable Version")
 SET(TensorFlow_ABI ${TensorFlow_ABI} CACHE STRING "The Python executable Version")
 SET(TensorFlow_LIBRARY ${TensorFlow_LIBRARY} CACHE PATH "The Python executable Version")
-SET(TensorFlow_FOUND ${TensorFlow_FOUND} CACHE BOOL "The Python executable Version")
+SET(TensorFlow_FOUND ${TensorFlow_FOUND} CACHE BOOL "The Python executable Version")

user_ops/__init__.py

@@ -41,9 +41,8 @@ flex_pool_grad = _flex_pooling_op_so.flex_pool_grad
 flex_deconv = _flex_deconvolution_op_so.flex_deconv
 flex_deconv_grad = _flex_deconvolution_op_so.flex_deconv_grad
 
+
 # pylint: disable=redefined-builtin
-
-
 def flex_convolution(features,
                      position,
                      neighborhood,