HomeSort by relevance Sort by last modified time
    Searched refs:XLACompileContext (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/compiler/
xla.py 101 class XLACompileContext(control_flow_ops.XLAControlFlowContext):
106 The primary role of `XLACompileContext` is to mark operators inside a
119 """Builds a new XLACompileContext.
124 pivot: a pivot node. Nodes in the XLACompileContext that do not have any
129 super(XLACompileContext, self).__init__()
172 """Create op in XLACompileContext and notifies outer context recursively."""
268 # Define the gradient loop state associated with the XLACompileContext to
269 # be None as the XLACompileContext does not get nested nor does the
270 # grad_state outside the XLACompileContext affect the graph inside so the
318 context = XLACompileContext(name=cluster_name, pivot=pivot
    [all...]
xla_test.py 56 return xla.XLACompileContext(name=computation_name, pivot=pivot)
169 """Tests that XLACompileContext is recognized as an XLA context."""
179 """Tests that ops created inside XLACompileContext can not be fed."""
187 """Tests that ops created inside XLACompileContext can not be fetched."""

Completed in 1169 milliseconds