Searched refs:XLA_TpuNodeContext (Results 1 – 3 of 3) sorted by relevance
50 explicit TpuNodeContext(int device_ordinal, XLA_TpuNodeContext* node_context) in TpuNodeContext()72 XLA_TpuNodeContext* const node_context_;
32 XLA_TpuNodeContext* node_context = in Create()
85 typedef struct XLA_TpuNodeContext XLA_TpuNodeContext; typedef419 XLA_TpuNodeContext* TpuNodeContext_Create(int device_ordinal,421 void TpuNodeContext_Free(XLA_TpuNodeContext* node_context);