Searched refs:XLACompileContext (Results 1 – 2 of 2) sorted by relevance
130 class XLACompileContext(control_flow_ops.XLAControlFlowContext): class158 super(XLACompileContext, self).__init__()347 context = XLACompileContext(name=cluster_name, pivot=pivot)
52 return xla.XLACompileContext(name=computation_name, pivot=pivot)