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

  /external/tensorflow/tensorflow/python/tpu/
tpu_config.py 55 'input_partition_dims',
94 input_partition_dims: A nested list to describe the partition dims
96 input_partition_dims must match the structure of `features` and
100 input_partition_dims = [[1, 2, 2, 1], None] will split the images to 4
123 input_partition_dims=None,
134 if input_partition_dims is not None:
135 if len(input_partition_dims) != 1 and len(input_partition_dims) != 2:
137 'input_partition_dims must be a list/tuple with one or two'
142 'input_partition_dims is only supported in PER_HOST_V2 mode.'
    [all...]
tpu_context.py 334 def input_partition_dims(self): member in class:_InternalTPUContext
335 return self._config.tpu_config.input_partition_dims
tpu_feed.py 737 input_partition_dims: A nested list/tuple of integers. Each inner
748 input_partition_dims=None,
758 self._input_partition_dims = input_partition_dims
    [all...]
tpu_estimator.py     [all...]

Completed in 139 milliseconds