Home
last modified time | relevance | path

Searched refs:TensorToDataHandle (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/csrc/lazy/core/
Dtensor_util.cpp41 BackendDataPtr TensorToDataHandle( in TensorToDataHandle() function
55 result.push_back(TensorToDataHandle(tensors[i], devices[i])); in CreateTensorsData()
Dtensor_util.h24 TensorToDataHandle(const at::Tensor& tensor, const BackendDevice& device);
Dtensor.cpp126 data()->handle = TensorToDataHandle(*data()->tensor_data, GetDevice()); in GetDataHandle()
232 data = TensorToDataHandle(tensor, device); in GetIrValueForTensor()
Dlazy_graph_executor.cpp198 BackendDataPtr device_data = TensorToDataHandle(tensor, device); in IrValueFromScalar()
296 device_data = TensorToDataHandle(tensor_copy, device); in GetDeviceData()