Searched refs:CopyToDeviceNode (Results 1 – 2 of 2) sorted by relevance
26 class CopyToDeviceNode : public EagerNode {28 CopyToDeviceNode(TensorHandle* src, Device* dstd, EagerContext* ctx) in CopyToDeviceNode() function38 ~CopyToDeviceNode() override { in ~CopyToDeviceNode()
1041 CopyToDeviceNode* node = new CopyToDeviceNode(h, dstd, ctx); in LocalEagerCopyToDevice()