Searched refs:_batch_tile_tensor (Results 1 – 1 of 1) sorted by relevance
878 static Tensor _batch_tile_tensor(const Tensor& self, IntArrayRef blocksize, const int64_t dense_dim… in _batch_tile_tensor() function1053 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()