Searched defs:oshape (Results 1 – 3 of 3) sorted by relevance
25 c10::SmallVector<int64_t, 3u> oshape; member
1429 auto oshape = output.sizes(); in upsample_generic_Nd_kernel_impl() local1526 auto oshape = output.sizes(); in _separable_upsample_generic_Nd_kernel_impl_single_dim() local
3481 oshape = [1, 400, 400, 256] variable