TGLite
TGLite
Getting Started
Quickstart
How does TBlock work?
API
tglite
tglite.TBatch
tglite.TBlock
tglite.TContext
tglite.TGraph
TGraph
Query graph structure
Query graph data
Query memory-based TGN data
Set device
tglite.TGraph.storage_device
tglite.TGraph.compute_device
tglite.TGraph.set_compute
tglite.TGraph.move_data
TGraph.move_data()
tglite.TSampler
tglite.Mailbox
tglite.Memory
tglite.nn
tglite.op
TGLite
Docs
»
tglite.TGraph
»
tglite.TGraph.move_data
Edit on GitHub
tglite.TGraph.move_data
¶
TGraph.
move_data
(
device
,
**
kwargs
)
¶
Moves tensor data to device while keeping graph on CPU