Searched defs:group_key (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 24 def all_reduce(t, group_size, group_key, instance_key, merge_op, final_op, argument
61 def all_gather(t, group_size, group_key, instance_key): argument
90 def broadcast_send(t, shape, dtype, group_size, group_key, instance_key): argument
137 def broadcast_recv(shape, dtype, group_size, group_key, instance_key): argument
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 499 int32 group_key = 1; field
507 int32 group_key = 1; field
524 int32 group_key = 5; field
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer_test.cc | 117 const int group_key = 5; in Init() local
|
| D | set_kernels.cc | 277 const auto group_key = group.group(); in Compute() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed.cc | 209 bool CollectiveParamResolverDistributed::GroupIsCached(int32 group_key) { in GroupIsCached()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 53 int32 group_key; member
|
| /external/wpa_supplicant_8/src/common/ |
| D | dpp.c | 528 static EVP_PKEY * dpp_set_pubkey_point(EVP_PKEY *group_key, in dpp_set_pubkey_point()
|