Searched refs:broadcast_send (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_CollectiveBcastSend.pbtxt | 4 name: "collective.broadcast_send"
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_multi_worker_test.py | 306 collective_ops.broadcast_send(in_tensor, (1,), dtypes.float32,
312 collective_ops.broadcast_send(in_tensor, (1,), dtypes.float32,
323 collective_ops.broadcast_send(in_tensor, (1,), dtypes.float32,
|
| D | collective_ops_test.py | 49 broadcast_send = _collective_ops.broadcast_send variable in CollectiveOpsV1
72 def broadcast_send(t, shape, dtype, group_size, group_key, instance_key, member in CollectiveOpsV2
232 collective_ops.broadcast_send(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops_gpu_test.py | 157 collectives.append(collective_ops.broadcast_send(
198 collective_ops.broadcast_send(
|
| D | collective_ops.py | 203 def broadcast_send(t, function
|
| D | collective_ops_test.py | 310 out0 = collective_ops.broadcast_send(in0, in0.shape, in0.dtype,
|
| /external/rust/crates/tokio/src/sync/tests/ |
| D | loom_broadcast.rs | 10 fn broadcast_send() { in broadcast_send() function
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | collective_all_reduce_strategy.py | 565 bcast_send = collective_ops.broadcast_send(
|
| /external/tensorflow/tensorflow/python/distribute/parallel_device/ |
| D | parallel_device_test.py | 248 s0 = collective_ops.broadcast_send(
|