Searched refs:node_context_ (Results 1 – 4 of 4) sorted by relevance
44 xla::Backend* backend() { return node_context_->backend(); } in backend()56 std::unique_ptr<tpu::TpuNodeContext> node_context_; variable
38 TF_ASSIGN_OR_RETURN(node_context_, in Acquire()
51 : device_ordinal_(device_ordinal), node_context_(node_context) { in TpuNodeContext()72 XLA_TpuNodeContext* const node_context_; variable
44 tpu::OpsApiFn()->TpuNodeContext_FreeFn(node_context_); in ~TpuNodeContext()