Searched defs:act_ndhwc (Results 1 – 1 of 1) sorted by relevance
288 const at::Tensor act_ndhwc = kSpatialDim == 2 in apply_impl() local865 const at::Tensor act_ndhwc = act.contiguous(channels_last); in apply_impl() local