Searched refs:getWriteableTensorData (Results 1 – 7 of 7) sorted by relevance
113 getWriteableTensorData(const at::Tensor& tensor, bool to_cpu);287 getWriteableTensorData(const at::Tensor& tensor, bool to_cpu = true);
720 WriteableTensorData getWriteableTensorData( in getWriteableTensorData() function
555 WriteableTensorData writable_td = getWriteableTensorData(td); in writeArchive()
81 WriteableTensorData writable_td = getWriteableTensorData(td); in writeArchiveAndTensors()
124 WriteableTensorData writable_td = getWriteableTensorData(td); in write_archive_current()347 WriteableTensorData writable_td = getWriteableTensorData(td); in backport_v5_to_v4()
62 WriteableTensorData writable_td = getWriteableTensorData(td); in writeArchive()
388 auto writeableTensorData = jit::getWriteableTensorData(tensorData[i]); in wireSerialize()