Searched defs:output_numel (Results 1 – 3 of 3) sorted by relevance
313 const int64_t output_numel = nbatch * num_kernels; in col2im_batched() local
302 const int output_numel = output_sizes_1 * output_sizes_2; in add_padding_2() local348 const int output_numel = output_sizes_1 * output_sizes_2 * output_sizes_3; in add_padding_3() local
2108 int64_t output_numel = 0; in allgather_coalesced() local