Home
last modified time | relevance | path

Searched refs:OpenRegTensorData (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/
H A D_device_daemon.py6 OpenRegTensorData,
148 res = OpenRegTensorData.from_meta(allocator, args[0])
152 dev_tensor = OpenRegTensorData.from_meta(allocator, dev_mem)
H A D_meta_parser.py27 class OpenRegTensorData(torch.Tensor): class
30 return OpenRegTensorData(allocator.tensor_from_meta(tensor_meta))