Skip to main content

build_activation_fwd_graph

atomr_accel_cuda::kernel::cudnn::activation

Function build_activation_fwd_graph 

Source 
pub fn build_activation_fwd_graph(
    dtype: DtypeTag,
    dims: &[i64],
    layout: TensorLayout,
    kind: ActivationKind,
) -> OperationGraphSpec
Expand description

Build the spec-side activation-fwd op graph.