Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dpickler.h97 struct WriteableTensorData { struct
112 friend TORCH_API WriteableTensorData argument
286 TORCH_API WriteableTensorData
H A Dpickler.cpp720 WriteableTensorData getWriteableTensorData( in getWriteableTensorData()
723 WriteableTensorData result; in getWriteableTensorData()
H A Dexport_module.cpp555 WriteableTensorData writable_td = getWriteableTensorData(td); in writeArchive()
H A Dexport.cpp81 WriteableTensorData writable_td = getWriteableTensorData(td); in writeArchiveAndTensors()
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/compatibility/
H A Dbackport_manager.cpp124 WriteableTensorData writable_td = getWriteableTensorData(td); in write_archive_current()
347 WriteableTensorData writable_td = getWriteableTensorData(td); in backport_v5_to_v4()
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/train/
H A Dexport_data.cpp62 WriteableTensorData writable_td = getWriteableTensorData(td); in writeArchive()