Searched refs:_gather (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/python/ops/ragged/ |
D | ragged_gather_ops.py | 111 return _gather(params, indices, axis, batch_dims) 114 def _gather(params, indices, axis, batch_dims): function 144 return indices.with_values(_gather(params, indices.values, 0, 0)) 217 _gather(params.values, indices.values, axis - 1, batch_dims - 1)) 232 adjusted_indices = _gather( 255 return _gather(flat_params, adjusted_indices, axis - 1, 0) 274 return params.with_values(_gather(params.values, indices, axis - 1, 0)) 295 return _gather(flat_params, adjusted_indices, axis - 1, 0)
|
/external/tensorflow/tensorflow/python/distribute/ |
D | test_util.py | 51 return nest.map_structure(functools.partial(_gather, strategy), value) 54 def _gather(strategy, value): function
|
D | cross_device_ops.py | 303 def _gather(self, per_replica_value, destinations, axis, options=None): member in CrossDeviceOps 923 …return ReductionToOneDevice()._gather(per_replica_value, destinations, axis, # pylint: disable=pr…
|
D | parameter_server_strategy.py | 509 return self._cross_device_ops._gather( # pylint: disable=protected-access
|
D | collective_all_reduce_strategy.py | 743 return self._get_cross_device_ops(value)._gather( # pylint: disable=protected-access
|
D | mirrored_strategy.py | 713 return self._get_cross_device_ops(value)._gather( # pylint: disable=protected-access
|
D | cross_device_ops_test.py | 651 gathered_values = collective._gather(
|
/external/tensorflow/tensorflow/ |
D | tensorflow.bzl | 1692 _transitive_hdrs(name = name + "_gather", deps = deps) 1693 native.filegroup(name = name, srcs = [":" + name + "_gather"]) 1749 name = name + "_gather", 1760 hdrs = [":" + name + "_gather"],
|
/external/mesa3d/docs/gallium/ |
D | tgsi.rst | 991 dst = texture\_gather (uint, coord, compare)
|