Searched defs:TpuNodeContext (Results 1 – 2 of 2) sorted by relevance
50 explicit TpuNodeContext(int device_ordinal, XLA_TpuNodeContext* node_context) in TpuNodeContext() function
67 explicit HostTransferManager(TpuNodeContext*, xla::Backend*) {} in HostTransferManager() argument