Searched defs:_gather (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | test_util.py | 83 def _gather(strategy, value): function
|
| H A D | cross_device_ops.py | 320 def _gather(self, per_replica_value, destinations, axis, options=None): member in CrossDeviceOps
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_gather_ops.py | 112 def _gather(params, indices, axis, batch_dims): function
|
| /aosp_15_r20/external/python/cpython3/Lib/test/test_asyncio/ |
| D | test_ssl.py | 366 async def _gather(*tasks): function
1230 async def _gather(*tasks): function
|
| D | test_tasks.py | 2973 def _gather(self, *args, **kwargs): member in FutureGatherTests
3070 def _gather(self, *args, **kwargs): member in CoroutineGatherTests
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 7124 def _gather(shape, index_dim, max_indices): function
7152 def _gather(shape, index_dim, max_indices): function
|