Searched defs:batched_out (Results 1 – 2 of 2) sorted by relevance
358 Tensor batched_out = batched_x.sum(0); in TEST() local368 Tensor batched_out = batched_x.sum(-1); in TEST() local378 Tensor batched_out = batched_x.sum(std::vector<int64_t>{0, 1}); in TEST() local388 Tensor batched_out = batched_x.sum(std::vector<int64_t>{0, 1}); in TEST() local696 auto batched_out = batched.expand({3, 5}); in TEST() local713 auto batched_out = batched.expand({7, 3, 5}); in TEST() local730 auto batched_out = batched.expand(c10::IntArrayRef({})); in TEST() local739 auto batched_out = batched.expand({5, 7}); in TEST() local752 auto batched_out = batched.unsqueeze(0); in TEST() local762 auto batched_out = batched.unsqueeze(0); in TEST() local[all …]
145 auto batched_out = at::zeros( in compressed_to_batched_compressed_indices() local