Searched defs:stride_arg (Results 1 – 2 of 2) sorted by relevance
90 IntArrayRef stride_arg, in pool2d()231 IntArrayRef stride_arg, in avg_pool2d()249 IntArrayRef stride_arg, in max_pool2d()
921 const IntArrayRef stride_arg, in Conv2dPackedContext()1256 const IntArrayRef stride_arg, in create()1330 const IntArrayRef stride_arg, in Conv1dPackedContext()