Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorConversions.cpp878 static Tensor _batch_tile_tensor(const Tensor& self, IntArrayRef blocksize, const int64_t dense_dim… in _batch_tile_tensor() function
1053 auto values = blocked_layout ? _batch_tile_tensor(self, blocksize, dense_dim) : self; in dense_to_sparse_compressed()
1054 …auto not_zero_mask = blocked_layout ? _batch_tile_tensor(self_mask, blocksize, dense_dim) : self_m… in dense_to_sparse_compressed()