tensor-safe-0.1.0.1: src/TensorSafe/Layers/GlobalAvgPooling2D.hs

{-| This module declares the GlobalAvgPooling2D layer data type. -}
module TensorSafe.Layers.GlobalAvgPooling2D (GlobalAvgPooling2D) where

import           Data.Map

import           TensorSafe.Compile.Expr
import           TensorSafe.Layer

-- | A GlobalAvgPooling2D function
data GlobalAvgPooling2D = GlobalAvgPooling2D deriving Show

instance Layer GlobalAvgPooling2D where
    layer = GlobalAvgPooling2D
    compile _ _ = CNLayer DGlobalAvgPooling2D empty