| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 25 group_key, argument
76 group_key, argument
125 group_key, argument
164 group_key, argument
207 group_key, argument
266 group_key, argument
301 group_key, argument
344 group_key, argument
|
| D | collective_ops_gpu_test.py | 254 def run_all_reduce(group_key, instance_key, merge_op): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_test.py | 56 def all_reduce(t, group_size, group_key, instance_key, *args, **kwargs): argument
64 def all_gather(t, group_size, group_key, instance_key, *args, **kwargs): argument
72 def broadcast_send(t, shape, dtype, group_size, group_key, instance_key, argument
81 def broadcast_recv(shape, dtype, group_size, group_key, instance_key, *args, argument
361 def run_all_reduce(group_size, group_key): argument
377 def run_and_assert(group_size, group_key): argument
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_param_resolver_local_test.cc | 303 CollectiveParams* MakeCollectiveParams(int group_key, int instance_key, in MakeCollectiveParams()
347 int group_key = 100; in TEST_F() local
396 int group_key = 100; in TEST_F() local
422 auto complete_params = [this, &cancel_mgr](int group_key, int instance_key) { in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_ops.cc | 28 static string CollectiveKey(OpKernelContext* ctx, int32 group_key, in CollectiveKey()
510 const Tensor& group_key = c->input(2); in ComputeAsync() local
639 const Tensor& group_key = c->input(2); in ComputeAsync() local
773 const Tensor& group_key = c->input(2); in ComputeAsync() local
898 const Tensor& group_key = c->input(1); in ComputeAsync() local
|
| D | set_kernels.cc | 277 const auto group_key = group.group(); in Compute() local
|
| D | collective_nccl_test.cc | 133 const int group_key = num_ranks; in Init() local
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 545 int32 group_key = 1; field
556 int32 group_key = 1; field
575 int32 group_key = 5; field
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 219 def get_instance_key(self, group_key, device): argument
264 group_key, argument
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed.cc | 223 CollectiveParamResolverDistributed::GetCachedGroup(int32 group_key) { in GetCachedGroup()
330 bool CollectiveParamResolverDistributed::InstanceIsCached(int32 group_key, in InstanceIsCached()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 63 int32 group_key; member
|
| /external/rust/crates/itertools/src/ |
| D | groupbylazy.rs | 241 fn group_key(&mut self, client: usize) -> K { in group_key() function
|
| /external/wpa_supplicant_8/src/common/ |
| D | dpp_crypto.c | 496 EVP_PKEY * dpp_set_pubkey_point(EVP_PKEY *group_key, const u8 *buf, size_t len) in dpp_set_pubkey_point()
|