Searched refs:expanded_size (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/crosvm/disk/src/ |
| H A D | android_sparse.rs | 98 expanded_size: u64, field
159 let expanded_size = chunk_header.chunk_sz.to_native() as u64 * blk_sz; in parse_chunk() localVariable
162 expanded_size, in parse_chunk()
208 let size = chunk_with_size.expanded_size; in from_parts()
225 let calculated_len: u64 = image.chunks.iter().map(|x| x.1.expanded_size).sum(); in from_parts()
265 expanded_size, in read_at_volatile()
274 let chunk_size = *expanded_size; in read_at_volatile()
386 expanded_size, in read_to_mem()
393 let chunk_size = *expanded_size; in read_to_mem()
480 expanded_size: 123, in parse_raw()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | memory_overlapping_test.cpp | 44 std::vector<int64_t> expanded_size(transpose_size); in TEST() local
45 expanded_size.insert(expanded_size.begin(), size_to_add); in TEST()
46 auto expanded = t.expand(expanded_size); in TEST()
72 std::vector<int64_t> expanded_size(size); in TEST() local
73 expanded_size.insert(expanded_size.begin(), size_to_add); in TEST()
74 auto expanded = t.expand(expanded_size); in TEST()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | ExpandUtils.h | 197 auto expanded_size = infer_size_symdimvector(s1, s2); in expand_outplace() local
199 c10::MaybeOwned<Tensor>::owned(to_expand1.expand_symint(expanded_size)), in expand_outplace()
200 c10::MaybeOwned<Tensor>::owned(to_expand2.expand_symint(expanded_size))); in expand_outplace()
253 auto expanded_size = in expand_outplace() local
256 c10::MaybeOwned<Tensor>::owned(to_expand1.expand(expanded_size)), in expand_outplace()
257 c10::MaybeOwned<Tensor>::owned(to_expand2.expand(expanded_size)), in expand_outplace()
258 c10::MaybeOwned<Tensor>::owned(to_expand3.expand(expanded_size))); in expand_outplace()
|
| H A D | LegacyVmapTransforms.cpp | 227 VmapDimVector expanded_size(batch_sizes.begin(), batch_sizes.end()); in logicalToPhysical() local
229 expanded_size.insert( in logicalToPhysical()
230 expanded_size.end(), in logicalToPhysical()
233 result.emplace_back(physical_tensor.expand(expanded_size), collective_levels); in logicalToPhysical()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Indexing.cu | 444 auto expanded_size = at::DimVector(expandedValue.sizes()); in index_put_with_sort_kernel() local
448 expanded_size = infer_size_dimvector(size1, size2); in index_put_with_sort_kernel()
451 expanded_size.insert(expanded_size.begin(), nElemBefore); in index_put_with_sort_kernel()
454 expanded_size.insert(expanded_size.end(), sliceSize); in index_put_with_sort_kernel()
456 expandedValue = expandedValue.expand(expanded_size); in index_put_with_sort_kernel()
577 auto expanded_size = at::DimVector(expandedValue.sizes()); in index_put_with_sort_quantized() local
581 expanded_size = infer_size_dimvector(size1, size2); in index_put_with_sort_quantized()
584 expanded_size.insert(expanded_size.begin(), nElemBefore); in index_put_with_sort_quantized()
586 expandedValue = expandedValue.expand(expanded_size); in index_put_with_sort_quantized()
|