Home
last modified time | relevance | path

Searched refs:_gather (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_gather_ops.py111 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/
Dtest_util.py51 return nest.map_structure(functools.partial(_gather, strategy), value)
54 def _gather(strategy, value): function
Dcross_device_ops.py303 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…
Dparameter_server_strategy.py509 return self._cross_device_ops._gather( # pylint: disable=protected-access
Dcollective_all_reduce_strategy.py743 return self._get_cross_device_ops(value)._gather( # pylint: disable=protected-access
Dmirrored_strategy.py713 return self._get_cross_device_ops(value)._gather( # pylint: disable=protected-access
Dcross_device_ops_test.py651 gathered_values = collective._gather(
/external/tensorflow/tensorflow/
Dtensorflow.bzl1692 _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/
Dtgsi.rst991 dst = texture\_gather (uint, coord, compare)