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

  /external/tensorflow/tensorflow/python/tpu/
tpu_feed.py 550 placement_function=None):
572 placement_function: if not None, a function that takes the shard index as
605 device=placement_function(index) if placement_function else None)
623 placement_function=None,
635 the shard. The Op for shard i is placed on device placement_function(i).
646 device_assignment is not `None`, but `placement_function` and
651 placement_function: if not None, a function that takes the shard
653 device the shard's infeed should be placed on. If placement_function
658 the shard's infeed should be placed on. If placement_function
    [all...]
tpu_estimator.py     [all...]

Completed in 458 milliseconds