Searched refs:src_contig (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Copy.cu | 302 Tensor src_contig; in copy_kernel_cuda() local
311 src_contig = iter.tensor(1).to(iter.dtype(0)).expand_as(dst).contiguous(); in copy_kernel_cuda()
315 src_contig = iter.tensor(1).expand_as(dst).contiguous(); in copy_kernel_cuda()
320 src_contig._set_conj(iter.tensor(1).is_conj()); in copy_kernel_cuda()
323 src_contig._set_neg(iter.tensor(1).is_neg()); in copy_kernel_cuda()
326 TORCH_INTERNAL_ASSERT(dst_contig.sizes().equals(src_contig.sizes())); in copy_kernel_cuda()
327 TORCH_INTERNAL_ASSERT(dst_contig.scalar_type() == src_contig.scalar_type()); in copy_kernel_cuda()
328 dst_contig.copy_(src_contig, non_blocking); in copy_kernel_cuda()
|
| /external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| D | Copy.cpp | 156 Tensor src_contig = src.contiguous(src.suggest_memory_format()); in pack_cpu_to_vulkan() local
173 memcpy_to_mapping(src_contig.to(at::kFloat), mapping); in pack_cpu_to_vulkan()
175 memcpy_to_mapping(src_contig, mapping); in pack_cpu_to_vulkan()
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | fbgemm_utils.cpp | 241 const Tensor src_contig = src.contiguous(); in ConvertToChannelsLast3dTensor() local
248 src_contig.data_ptr<scalar_t>(), in ConvertToChannelsLast3dTensor()
356 const Tensor src_contig = src.contiguous(); in ConvertConvWeightsToChannelLastTensor() local
364 src_contig.data_ptr<scalar_t>(), in ConvertConvWeightsToChannelLastTensor()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | EmbeddingBag.cpp | 206 auto src_contig = src.contiguous(); in index_select_add() local
207 auto* src_data = src_contig.const_data_ptr<data_t>(); in index_select_add()
384 auto src_contig = src.contiguous(); in index_select_add() local
385 auto* src_data = src_contig.const_data_ptr<float>(); in index_select_add()
572 auto src_contig = src.contiguous(); in index_select_scale_add() local
573 auto* src_data = src_contig.const_data_ptr<data_t>(); in index_select_scale_add()
761 auto src_contig = src.contiguous(); in index_select_scale_add() local
762 auto* src_data = src_contig.const_data_ptr<float>(); in index_select_scale_add()
|
| /external/pytorch/test/ |
| D | test_torch.py | 3410 for dest_contig, src_contig, index_contig in product([True, False], repeat=3):
3414 src = make_arg(other_sizes, num_copy, dim, src_contig)
3574 for src_contig, idx_contig in product([True, False], repeat=2):
3577 src = make_arg(other_sizes, num_src, dim, src_contig)
3618 for src_contig, idx_contig, idx_reshape in product([True, False], repeat=3):
3620 src = make_arg(src_size, noncontiguous=not src_contig)
3652 …for dst_contig, src_contig, idx_contig, idx_reshape, accumulate in product([True, False], repeat=5…
3655 src = make_arg(src_size, noncontiguous=not src_contig)
6820 for dest_contig, src_contig, index_contig in product([True, False], repeat=3):
6828 if not src_contig:
|