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. If the function returns m outputs, rewrite will return a list of m 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.