Searched defs:lazy_tensor (Results 1 – 4 of 4) sorted by relevance
24 auto lazy_tensor = TryGetLtcTensor(tensor); in GetOrCreateLtcTensor() local37 LazyTensorPtr lazy_tensor = in Create() local44 LazyTensorPtr lazy_tensor = in Create() local51 LazyTensorPtr lazy_tensor = in Create() local360 auto lazy_tensor = TryGetLtcTensor(tensor); in GetLtcTensor() local
91 void LTCTensorImpl::set_tensor(const LazyTensorPtr& lazy_tensor) { in set_tensor()
39 torch::lazy::LazyTensorPtr lazy_tensor = torch::lazy::TryGetLtcTensor(tensor); in GetTensorId() local51 torch::lazy::LazyTensorPtr lazy_tensor = in GetTensorsDump() local70 auto lazy_tensor = torch::lazy::TryGetLtcTensor(tensor); in GetLtcTensors() local
118 torch::lazy::LazyTensorPtr lazy_tensor = torch::lazy::TryGetLtcTensor(tensor); in GetTensorTextGraph() local123 torch::lazy::LazyTensorPtr lazy_tensor = torch::lazy::TryGetLtcTensor(tensor); in GetTensorDotGraph() local