Searched refs:ordering_token (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 82 ordering_token=None): argument
109 if ordering_token is not None:
110 ordering_token = [ordering_token]
120 ordering_token=ordering_token or [])
168 ordering_token=None): argument
191 if ordering_token is not None:
192 ordering_token = [ordering_token]
200 ordering_token=ordering_token or [])
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 354 ordering_token = self._get_ordering_token(communication_hint)
364 ordering_token=ordering_token)
379 ordering_token = self._get_ordering_token(communication_hint)
388 ordering_token=ordering_token)
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CollectiveGatherV2.pbtxt | 71 name: "ordering_token"
|
| D | CollectiveReduceV2.pbtxt | 159 name: "ordering_token"
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_test.py | 971 ordering_token=token0.handle)
978 ordering_token=token1.handle)
991 ordering_token=token0.handle)
998 ordering_token=token1.handle)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 773 …argspec: "args=[\'input\', \'group_size\', \'group_key\', \'instance_key\', \'ordering_token\', \'…
785 …argspec: "args=[\'input\', \'group_size\', \'group_key\', \'instance_key\', \'ordering_token\', \'…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 773 …argspec: "args=[\'input\', \'group_size\', \'group_key\', \'instance_key\', \'ordering_token\', \'…
785 …argspec: "args=[\'input\', \'group_size\', \'group_key\', \'instance_key\', \'ordering_token\', \'…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 19302 …, group_size tf.Output, group_key tf.Output, instance_key tf.Output, ordering_token []tf.Output, m…
19313 input, group_size, group_key, instance_key, tf.OutputList(ordering_token),
50505 …, group_size tf.Output, group_key tf.Output, instance_key tf.Output, ordering_token []tf.Output, o…
50516 input, group_size, group_key, instance_key, tf.OutputList(ordering_token),
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 7693 name: "ordering_token"
7882 name: "ordering_token"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 1929 Variadic<TF_ResourceTensor>:$ordering_token,
|