Searched refs:all_contiguous (Results 1 – 7 of 7) sorted by relevance
94 bool all_contiguous = in tensors_have_same_dim_order() local106 all_contiguous = all_contiguous && in tensors_have_same_dim_order()113 all_contiguous || all_channels_last, in tensors_have_same_dim_order()117 return all_contiguous || all_channels_last; in tensors_have_same_dim_order()
115 bool all_contiguous = true; in tensors_have_same_dim_order() local118 all_contiguous = all_contiguous && in tensors_have_same_dim_order()129 all_contiguous || all_channels_last, in tensors_have_same_dim_order()
141 bool all_contiguous = is_contiguous(input) in batch_norm_cpu_transform_input_template() local149 if (all_contiguous) { in batch_norm_cpu_transform_input_template()210 bool all_contiguous = is_contiguous(input); in batch_norm_cpu_update_stats_template() local220 if (all_contiguous) { in batch_norm_cpu_update_stats_template()328 bool all_contiguous = is_contiguous(input) in batch_norm_backward_cpu_template() local332 if (all_contiguous) { in batch_norm_backward_cpu_template()816 const bool all_contiguous = is_contiguous(self) in batch_norm_cpu() local821 …Tensor output = at::empty_like(self, all_contiguous ? suggest_memory_format_contig(self) : self.su… in batch_norm_cpu()
269 bool all_contiguous = true; in TORCH_PRECOMPUTE_META_FUNC() local288 all_contiguous = result.is_contiguous(memory_format); in TORCH_PRECOMPUTE_META_FUNC()315 all_contiguous = all_contiguous && t.is_contiguous(memory_format); in TORCH_PRECOMPUTE_META_FUNC()320 all_contiguous = false; in TORCH_PRECOMPUTE_META_FUNC()344 .set_all_contiguous(all_contiguous) in TORCH_PRECOMPUTE_META_FUNC()581 bool all_contiguous, in TORCH_IMPL_FUNC()597 …if (use_serial_kernel && all_contiguous && all_same_dtype && (MemoryFormat::Contiguous == memory_f… in TORCH_IMPL_FUNC()606 if (use_serial_kernel && all_contiguous && all_same_dtype && serial_dtype) { in TORCH_IMPL_FUNC()
424 bool all_contiguous(TensorList tensors) { in all_contiguous() function780 if (detail::all_contiguous(tensors)) { in _chunk_cat_cuda()811 detail::all_contiguous(tensors) && out.is_non_overlapping_and_dense(); in _chunk_cat_out_cuda()
457 bool all_contiguous, in TORCH_IMPL_FUNC()494 all_contiguous && in TORCH_IMPL_FUNC()
144 bool all_contiguous,