Searched refs:lazy_tensor (Results 1 – 6 of 6) sorted by relevance
24 auto lazy_tensor = TryGetLtcTensor(tensor); in GetOrCreateLtcTensor() local25 return lazy_tensor ? lazy_tensor : LazyTensor::Create(tensor, device); in GetOrCreateLtcTensor()37 LazyTensorPtr lazy_tensor = in Create() local39 LazyGraphExecutor::Get()->RegisterTensor(lazy_tensor->data()); in Create()40 return lazy_tensor; in Create()44 LazyTensorPtr lazy_tensor = in Create() local46 LazyGraphExecutor::Get()->RegisterTensor(lazy_tensor->data()); in Create()47 return lazy_tensor; in Create()51 LazyTensorPtr lazy_tensor = in Create() local53 LazyGraphExecutor::Get()->RegisterTensor(lazy_tensor->data()); in Create()[all …]
91 void LTCTensorImpl::set_tensor(const LazyTensorPtr& lazy_tensor) { in set_tensor() argument92 tensor_ = c10::make_intrusive<LazyTensor>(*lazy_tensor); in set_tensor()
24 void set_tensor(const LazyTensorPtr& lazy_tensor);
39 torch::lazy::LazyTensorPtr lazy_tensor = torch::lazy::TryGetLtcTensor(tensor); in GetTensorId() local40 return lazy_tensor->GetUniqueId(); in GetTensorId()51 torch::lazy::LazyTensorPtr lazy_tensor = in GetTensorsDump() local53 values.push_back(lazy_tensor->GetIrValue()); in GetTensorsDump()70 auto lazy_tensor = torch::lazy::TryGetLtcTensor(tensor); in GetLtcTensors() local71 if (lazy_tensor) { in GetLtcTensors()72 lazy_tensors.push_back(lazy_tensor); in GetLtcTensors()
118 torch::lazy::LazyTensorPtr lazy_tensor = torch::lazy::TryGetLtcTensor(tensor); in GetTensorTextGraph() local119 return torch::lazy::DumpUtil::ToText({lazy_tensor->GetIrValue().node.get()}); in GetTensorTextGraph()123 torch::lazy::LazyTensorPtr lazy_tensor = torch::lazy::TryGetLtcTensor(tensor); in GetTensorDotGraph() local124 return torch::lazy::DumpUtil::ToDot({lazy_tensor->GetIrValue().node.get()}); in GetTensorDotGraph()
78 for dup_idx, lazy_tensor in enumerate(lazy_out_list):79 uniq_idx = tensor_id_to_idx.get(id(lazy_tensor), None)85 tensor_id_to_idx[id(lazy_tensor)] = uniq_idx