Skip to main content

build_pool_bwd_graph

atomr_accel_cuda::kernel::cudnn::pool

Function build_pool_bwd_graph 

Source 
pub fn build_pool_bwd_graph(
    dtype: DtypeTag,
    x_dims: &[i64],
    y_dims: &[i64],
    layout: TensorLayout,
    p: &PoolParams,
) -> OperationGraphSpec