Searched refs:xtensor (Results 1 – 4 of 4) sorted by relevance
229 auto xtensor = TryGetLtcTensor(tensor); in initLazyBindings() local230 roots.push_back(xtensor->GetIrValue().node.get()); in initLazyBindings()
83 Tensor xtensor = at::compositeexplicitautograd::expand(self, padded_size); in repeat_out() local85 for (const auto i : c10::irange(xtensor.dim())) { in repeat_out()89 i, xtensor.size(i), std::max<int64_t>(xtensor.size(i), 1)); in repeat_out()92 at::native::copy_(urtensor, xtensor.expand_as(urtensor)); in repeat_out()
1525 Tensor xtensor = self.expand(padded_size); in repeat() local1540 for (const auto i : c10::irange(xtensor.dim())) { in repeat()1543 auto size_i = xtensor.sizes()[i]; in repeat()1547 urtensor.copy_(xtensor.expand_as(urtensor)); in repeat()
2221 dep_xtensor = dependency('xtensor')