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

  /external/tensorflow/tensorflow/python/tpu/
tpu_feed.py 792 def generate_enqueue_ops(self, per_host_sharded_inputs):
795 per_host_sharded_inputs is a list, one for each replica, of lists of
810 per_host_sharded_inputs: a list of lists of Tensors. The length of th
    [all...]
tpu_estimator.py 740 per_host_sharded_inputs = []
761 per_host_sharded_inputs.append(flattened_inputs)
764 number_of_tuple_elements=len(per_host_sharded_inputs[0]))
768 per_host_sharded_inputs, tpu_ordinal_function=tpu_ordinal_function_impl)
    [all...]

Completed in 569 milliseconds