Home
last modified time | relevance | path

Searched refs:lazy_tensor (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dtensor.cpp24 auto lazy_tensor = TryGetLtcTensor(tensor); in GetOrCreateLtcTensor() local
25 return lazy_tensor ? lazy_tensor : LazyTensor::Create(tensor, device); in GetOrCreateLtcTensor()
37 LazyTensorPtr lazy_tensor = in Create() local
39 LazyGraphExecutor::Get()->RegisterTensor(lazy_tensor->data()); in Create()
40 return lazy_tensor; in Create()
44 LazyTensorPtr lazy_tensor = in Create() local
46 LazyGraphExecutor::Get()->RegisterTensor(lazy_tensor->data()); in Create()
47 return lazy_tensor; in Create()
51 LazyTensorPtr lazy_tensor = in Create() local
53 LazyGraphExecutor::Get()->RegisterTensor(lazy_tensor->data()); in Create()
[all …]
H A Dtensor_impl.cpp91 void LTCTensorImpl::set_tensor(const LazyTensorPtr& lazy_tensor) { in set_tensor() argument
92 tensor_ = c10::make_intrusive<LazyTensor>(*lazy_tensor); in set_tensor()
H A Dtensor_impl.h24 void set_tensor(const LazyTensorPtr& lazy_tensor);
/aosp_15_r20/external/pytorch/torch/csrc/lazy/python/
H A Dinit.cpp39 torch::lazy::LazyTensorPtr lazy_tensor = torch::lazy::TryGetLtcTensor(tensor); in GetTensorId() local
40 return lazy_tensor->GetUniqueId(); in GetTensorId()
51 torch::lazy::LazyTensorPtr lazy_tensor = in GetTensorsDump() local
53 values.push_back(lazy_tensor->GetIrValue()); in GetTensorsDump()
70 auto lazy_tensor = torch::lazy::TryGetLtcTensor(tensor); in GetLtcTensors() local
71 if (lazy_tensor) { in GetLtcTensors()
72 lazy_tensors.push_back(lazy_tensor); in GetLtcTensors()
/aosp_15_r20/external/pytorch/test/cpp/lazy/
H A Dtest_lazy_ops_util.cpp118 torch::lazy::LazyTensorPtr lazy_tensor = torch::lazy::TryGetLtcTensor(tensor); in GetTensorTextGraph() local
119 return torch::lazy::DumpUtil::ToText({lazy_tensor->GetIrValue().node.get()}); in GetTensorTextGraph()
123 torch::lazy::LazyTensorPtr lazy_tensor = torch::lazy::TryGetLtcTensor(tensor); in GetTensorDotGraph() local
124 return torch::lazy::DumpUtil::ToDot({lazy_tensor->GetIrValue().node.get()}); in GetTensorDotGraph()
/aosp_15_r20/external/pytorch/torch/_lazy/
H A Dextract_compiled_graph.py78 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