Home
last modified time | relevance | path

Searched refs:all_same_sizes_and_stride (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/
H A DShape.mm146 bool all_same_sizes_and_stride,
259 key += getTensorsStringKey(input_tensors, true, all_same_sizes_and_stride);
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp271 bool all_same_sizes_and_stride = true; in TORCH_PRECOMPUTE_META_FUNC() local
316 all_same_sizes_and_stride = all_same_sizes_and_stride && in TORCH_PRECOMPUTE_META_FUNC()
346 .set_all_same_sizes_and_stride(all_same_sizes_and_stride) in TORCH_PRECOMPUTE_META_FUNC()
583 bool all_same_sizes_and_stride, in TORCH_IMPL_FUNC()
612 if (all_same_sizes_and_stride && result.is_contiguous(memory_format) && in TORCH_IMPL_FUNC()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DShape.cu459 bool all_same_sizes_and_stride, in TORCH_IMPL_FUNC()