Searched refs:broadcast_recv (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/core/api_def/python_api/ |
D | api_def_CollectiveBcastRecv.pbtxt | 4 name: "collective.broadcast_recv"
|
/external/tensorflow/tensorflow/python/ops/ |
D | collective_ops_gpu_test.py | 161 collectives.append(collective_ops.broadcast_recv( 180 collective_ops.broadcast_recv(
|
D | collective_ops.py | 298 def broadcast_recv(shape, function
|
D | collective_ops_test.py | 314 out1 = collective_ops.broadcast_recv(c1.shape, c1.dtype,
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | collective_ops_test.py | 50 broadcast_recv = _collective_ops.broadcast_recv variable in CollectiveOpsV1 81 def broadcast_recv(shape, dtype, group_size, group_key, instance_key, *args, member in CollectiveOpsV2 242 collective_ops.broadcast_recv(
|
D | collective_ops_multi_worker_test.py | 326 collective_ops.broadcast_recv((1,), dtypes.float32, group_size,
|
/external/tensorflow/tensorflow/python/distribute/ |
D | collective_all_reduce_strategy.py | 571 return collective_ops.broadcast_recv(initial_value.shape,
|
/external/tensorflow/tensorflow/python/distribute/parallel_device/ |
D | parallel_device_test.py | 255 r0 = collective_ops.broadcast_recv(
|