/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()
|