Searched refs:GetDeviceData (Results 1 – 3 of 3) sorted by relevance
44 BackendDataPtr GetDeviceData(48 BackendDataPtr GetDeviceData(229 BackendDataPtr GetDeviceData(233 BackendDataPtr GetDeviceData(
288 BackendDataPtr LazyGraphExecutor::DataCacheArena::GetDeviceData( in GetDeviceData() function in torch::lazy::LazyGraphExecutor::DataCacheArena303 BackendDataPtr LazyGraphExecutor::DataCacheArena::GetDeviceData( in GetDeviceData() function in torch::lazy::LazyGraphExecutor::DataCacheArena316 return GetDeviceData(t, device); in GetDeviceData()384 BackendDataPtr LazyGraphExecutor::GetDeviceData( in GetDeviceData() function in torch::lazy::LazyGraphExecutor387 return DataCacheArena::Get()->GetDeviceData(tensor, device); in GetDeviceData()390 BackendDataPtr LazyGraphExecutor::GetDeviceData( in GetDeviceData() function in torch::lazy::LazyGraphExecutor394 return DataCacheArena::Get()->GetDeviceData(value, scalar_type, device); in GetDeviceData()486 BackendDataPtr data = GetDeviceData(value, type, device); in GetDeviceDataIrValue()498 auto data = GetDeviceData(value, value.type(), device); in GetIrValueForScalarFromCodegen()
228 data = LazyGraphExecutor::Get()->GetDeviceData(tensor.cpu(), device); in GetIrValueForTensor()