Home
last modified time | relevance | path

Searched defs:group_key (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dcollective_ops.py25 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
Dcollective_ops_gpu_test.py254 def run_all_reduce(group_key, instance_key, merge_op): argument
/external/tensorflow/tensorflow/python/kernel_tests/
Dcollective_ops_test.py56 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/
Dcollective_param_resolver_local_test.cc303 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/
Dcollective_ops.cc28 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
Dset_kernels.cc277 const auto group_key = group.group(); in Compute() local
Dcollective_nccl_test.cc133 const int group_key = num_ranks; in Init() local
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto545 int32 group_key = 1; field
556 int32 group_key = 1; field
575 int32 group_key = 5; field
/external/tensorflow/tensorflow/python/distribute/
Dcross_device_utils.py219 def get_instance_key(self, group_key, device): argument
264 group_key, argument
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed.cc223 CollectiveParamResolverDistributed::GetCachedGroup(int32 group_key) { in GetCachedGroup()
330 bool CollectiveParamResolverDistributed::InstanceIsCached(int32 group_key, in InstanceIsCached()
/external/tensorflow/tensorflow/core/framework/
Dcollective.h63 int32 group_key; member
/external/rust/crates/itertools/src/
Dgroupbylazy.rs241 fn group_key(&mut self, client: usize) -> K { in group_key() function
/external/wpa_supplicant_8/src/common/
Ddpp_crypto.c496 EVP_PKEY * dpp_set_pubkey_point(EVP_PKEY *group_key, const u8 *buf, size_t len) in dpp_set_pubkey_point()