Searched defs:expanded_size (Results 1 – 5 of 5) sorted by relevance
44 std::vector<int64_t> expanded_size(transpose_size); in TEST() local72 std::vector<int64_t> expanded_size(size); in TEST() local
197 auto expanded_size = infer_size_symdimvector(s1, s2); in expand_outplace() local253 auto expanded_size = in expand_outplace() local
227 VmapDimVector expanded_size(batch_sizes.begin(), batch_sizes.end()); in logicalToPhysical() local
98 expanded_size: u64, field159 let expanded_size = chunk_header.chunk_sz.to_native() as u64 * blk_sz; in parse_chunk() localVariable
444 auto expanded_size = at::DimVector(expandedValue.sizes()); in index_put_with_sort_kernel() local577 auto expanded_size = at::DimVector(expandedValue.sizes()); in index_put_with_sort_quantized() local