Searched defs:output_contig (Results 1 – 4 of 4) sorted by relevance
185 const Tensor& output_contig, in _segment_reduce_cpu_lengths_backward_kernel1()321 const Tensor& output_contig, in _segment_reduce_cpu_lengths_backward_kernel()354 const Tensor& output_contig, in _segment_reduce_cpu_offsets_backward_kernel()502 const auto output_contig = output.contiguous(); in _segment_reduce_backward_kernel() local
264 const Tensor& output_contig, in _segment_reduce_lengths_offsets_backward_cuda_kernel()366 const Tensor& output_contig, in _segment_reduce_lengths_backward_cuda_kernel()378 const Tensor& output_contig, in _segment_reduce_offsets_backward_cuda_kernel()
133 auto output_contig = output.expect_contiguous(); in qembeddingbag_byte_unpack_out() local
2715 auto output_contig = out.expect_contiguous(); in signed_log1p_out() local