• Alexander Matyasko's avatar
    Add Max3dPoolGradGrad and padding · 9411992c
    Alexander Matyasko 提交于
    Add Max3dPoolGradGrad which accept 5d input and pools over last 3
    dimensions. Also add padding parameter and now accept tensor variables
    to make_node which removes necessity to check for constant args. Local
    optimization are updated accordingly which now work for 2d and 3d
    pooling.
    9411992c
opt.py 69.3 KB