Home
last modified time | relevance | path

Searched refs:broadcast_recv (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_CollectiveBcastRecv.pbtxt4 name: "collective.broadcast_recv"
/external/tensorflow/tensorflow/python/ops/
Dcollective_ops_gpu_test.py161 collectives.append(collective_ops.broadcast_recv(
180 collective_ops.broadcast_recv(
Dcollective_ops.py298 def broadcast_recv(shape, function
Dcollective_ops_test.py314 out1 = collective_ops.broadcast_recv(c1.shape, c1.dtype,
/external/tensorflow/tensorflow/python/kernel_tests/
Dcollective_ops_test.py50 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(
Dcollective_ops_multi_worker_test.py326 collective_ops.broadcast_recv((1,), dtypes.float32, group_size,
/external/tensorflow/tensorflow/python/distribute/
Dcollective_all_reduce_strategy.py571 return collective_ops.broadcast_recv(initial_value.shape,
/external/tensorflow/tensorflow/python/distribute/parallel_device/
Dparallel_device_test.py255 r0 = collective_ops.broadcast_recv(