Home
last modified time | relevance | path

Searched refs:BackendDataPtr (Results 1 – 13 of 13) sorted by relevance

/external/pytorch/torch/csrc/lazy/core/
Dlazy_graph_executor.h44 BackendDataPtr GetDeviceData(
48 BackendDataPtr GetDeviceData(
161 std::vector<BackendDataPtr> parameters_data;
229 BackendDataPtr GetDeviceData(
233 BackendDataPtr GetDeviceData(
317 std::vector<BackendDataPtr> parameters_data,
318 std::vector<BackendDataPtr> tensors_data,
327 std::vector<BackendDataPtr> parameters_data;
330 std::vector<BackendDataPtr> tensors_data;
348 std::vector<BackendDataPtr> parameters_data;
[all …]
Dtensor.h28 Data(BackendDataPtr handle, BackendDevice device) in Data()
48 BackendDataPtr handle;
60 static LazyTensorPtr Create(BackendDataPtr handle);
106 BackendDataPtr GetDataHandle();
110 BackendDataPtr CurrentDataHandle() const;
112 void SetDataHandle(BackendDataPtr handle);
113 void SetDataHandle(BackendDataPtr handle, bool sync);
157 Value CreateTensorNode(BackendDataPtr data, bool read_only) const;
162 explicit LazyTensor(BackendDataPtr handle);
Dtensor_util.cpp31 c10::ArrayRef<BackendDataPtr> data_handles, in DataHandlesToTensors()
41 BackendDataPtr TensorToDataHandle( in TensorToDataHandle()
48 std::vector<BackendDataPtr> CreateTensorsData( in CreateTensorsData()
52 std::vector<BackendDataPtr> result; in CreateTensorsData()
Dlazy_graph_executor.cpp198 BackendDataPtr device_data = TensorToDataHandle(tensor, device); in IrValueFromScalar()
288 BackendDataPtr LazyGraphExecutor::DataCacheArena::GetDeviceData( in GetDeviceData()
293 BackendDataPtr device_data = cache->Get(tensor); in GetDeviceData()
303 BackendDataPtr LazyGraphExecutor::DataCacheArena::GetDeviceData( in GetDeviceData()
384 BackendDataPtr LazyGraphExecutor::GetDeviceData( in GetDeviceData()
390 BackendDataPtr LazyGraphExecutor::GetDeviceData( in GetDeviceData()
486 BackendDataPtr data = GetDeviceData(value, type, device); in GetDeviceDataIrValue()
538 std::vector<BackendDataPtr> parameters_data, in Async()
539 std::vector<BackendDataPtr> tensors_data, in Async()
624 std::vector<BackendDataPtr> handles = in CollectSyncTensors()
[all …]
Dtensor.cpp50 LazyTensorPtr LazyTensor::Create(BackendDataPtr handle) { in Create()
64 LazyTensor::LazyTensor(BackendDataPtr handle) in LazyTensor()
112 BackendDataPtr LazyTensor::GetDataHandle() { in GetDataHandle()
113 BackendDataPtr handle = CurrentDataHandle(); in GetDataHandle()
132 BackendDataPtr LazyTensor::CurrentDataHandle() const { in CurrentDataHandle()
136 void LazyTensor::SetDataHandle(BackendDataPtr handle) { in SetDataHandle()
140 void LazyTensor::SetDataHandle(BackendDataPtr handle, bool sync) { in SetDataHandle()
189 BackendDataPtr handle = CurrentDataHandle(); in GetIrValue()
221 BackendDataPtr data; in GetIrValueForTensor()
296 Value LazyTensor::CreateTensorNode(BackendDataPtr data, bool read_only) const { in CreateTensorNode()
Dtensor_util.h18 c10::ArrayRef<BackendDataPtr> data_handles,
23 TORCH_API BackendDataPtr
28 TORCH_API std::vector<BackendDataPtr> CreateTensorsData(
/external/pytorch/torch/csrc/lazy/backend/
Dbackend_interface.h49 virtual BackendDataPtr MakeComputationDataFromTensor(
53 virtual BackendDataPtr MakeComputationDataFromScalar(
56 virtual BackendDataPtr CreateDataPlaceholder(
62 virtual BackendDataPtr GetComputationDataFromNode(const Node*) const = 0;
65 const BackendDataPtr data,
90 virtual std::vector<BackendDataPtr> ExecuteComputation(
92 c10::ArrayRef<BackendDataPtr> arguments,
Dlowering_context.h66 const std::vector<BackendDataPtr>& GetParametersData() const;
79 const BackendDataPtr& parameter_data, in CheckResultShape()
108 std::vector<BackendDataPtr> parameters_;
Dlowering_context.cpp16 const std::vector<BackendDataPtr>& LoweringContext::GetParametersData() const { in GetParametersData()
Dbackend_data.h58 using BackendDataPtr = std::shared_ptr<BackendData>; variable
/external/pytorch/torch/csrc/lazy/ts_backend/
Dts_backend_impl.cpp83 const torch::lazy::BackendDataPtr data, in MakeTensorFromComputationData()
89 torch::lazy::BackendDataPtr MakeComputationDataFromTensor( in MakeComputationDataFromTensor()
110 torch::lazy::BackendDataPtr MakeComputationDataFromScalar( in MakeComputationDataFromScalar()
116 torch::lazy::BackendDataPtr GetComputationDataFromNode( in GetComputationDataFromNode()
135 torch::lazy::BackendDataPtr CreateDataPlaceholder(
142 std::vector<torch::lazy::BackendDataPtr> ExecuteComputation(
144 c10::ArrayRef<torch::lazy::BackendDataPtr> arguments,
189 torch::lazy::BackendDataPtr TSBackendImpl::CreateDataPlaceholder( in CreateDataPlaceholder()
207 std::vector<torch::lazy::BackendDataPtr> TSBackendImpl::ExecuteComputation( in ExecuteComputation()
209 c10::ArrayRef<torch::lazy::BackendDataPtr> arguments, in ExecuteComputation()
[all …]
Dts_lowering_context.cpp58 torch::jit::Value* TSLoweringContext::GetParameter(BackendDataPtr data) { in GetParameter()
Dts_lowering_context.h127 torch::jit::Value* GetParameter(BackendDataPtr data);