tf.contrib.tpu.rewrite( computation, inputs=None, infeed_queue=None, device_assignment=None, name=None )
computation for execution on a TPU system.
computation: A Python function that builds a computation to apply to the input. If the function takes n inputs, 'inputs' should be a list of n tensors.
computationmay return a list of operations and tensors. Tensors must come before operations in the returned list. The return value of
rewriteis a list of tensors corresponding to the tensors from the output of
Operations returned from
computationwill be executed when evaluating any of the returned output tensors.
inputs: A list of input tensors or
None(equivalent to an empty list).
infeed_queue: If not
InfeedQueuefrom which to append a tuple of arguments as inputs to
device_assignment: if not
DeviceAssignmentdescribing the mapping between logical cores in the computation with physical cores in the TPU topology. May be omitted for a single-core computation, in which case the core attached to task 0, TPU device 0 is used.
name: (Deprecated) Does nothing.
A list of output tensors.