Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/ops/
Dcollective_ops.py21 group_key, argument
97 group_key, argument
161 group_key, argument
201 group_key, argument
252 group_key, argument
314 group_key, argument
349 group_key, argument
393 group_key, argument
427 def initialize_communicator(group_key, argument
Dcollective_ops_gpu_test.py250 def run_all_reduce(group_key, instance_key, merge_op): argument
/external/cronet/net/reporting/
Dreporting_cache_impl.cc238 const ReportingEndpointGroupKey& group_key, in GetEndpointStats()
267 const ReportingEndpointGroupKey& group_key, in IncrementEndpointDeliveries()
525 const ReportingEndpointGroupKey& group_key) { in RemoveEndpointGroup()
561 const ReportingEndpointGroupKey& group_key = endpoint_it->first; in RemoveEndpointsForUrl() local
606 const ReportingEndpointGroupKey& group_key = group.group_key; in AddClientsLoadedFromStore() local
693 const ReportingEndpointGroupKey& group_key) { in GetCandidateEndpointsForDelivery()
805 const ReportingEndpointGroupKey& group_key, in GetEndpointForTesting()
817 const ReportingEndpointGroupKey& group_key, in EndpointGroupExistsForTesting()
861 const ReportingEndpointGroupKey& group_key, in SetV1EndpointForTesting()
898 const ReportingEndpointGroupKey& group_key, in SetEndpointForTesting()
[all …]
Dreporting_test_util.cc216 const ReportingEndpointGroupKey& group_key, in FindEndpointInCache()
222 const ReportingEndpointGroupKey& group_key, in SetEndpointInCache()
235 const ReportingEndpointGroupKey& group_key, in SetV1EndpointInCache()
244 const ReportingEndpointGroupKey& group_key, in EndpointExistsInCache()
251 const ReportingEndpointGroupKey& group_key, in GetEndpointStatistics()
266 const ReportingEndpointGroupKey& group_key, in EndpointGroupExistsInCache()
Dreporting_header_parser.cc73 const ReportingEndpointGroupKey& group_key, in ProcessEndpoint()
138 ReportingEndpointGroupKey group_key(network_anonymization_key, origin, in ProcessEndpointGroup() local
195 const ReportingEndpointGroupKey& group_key, in ProcessEndpoint()
226 ReportingEndpointGroupKey group_key(network_anonymization_key, in ProcessV1Endpoint() local
Dreporting_endpoint_manager_unittest.cc51 const ReportingEndpointGroupKey& group_key) override { in GetCandidateEndpointsForDelivery()
100 void IncrementEndpointDeliveries(const ReportingEndpointGroupKey& group_key, in IncrementEndpointDeliveries()
166 const ReportingEndpointGroupKey& group_key) override { in RemoveEndpointGroup()
196 const ReportingEndpointGroupKey& group_key, in GetEndpointForTesting()
201 bool EndpointGroupExistsForTesting(const ReportingEndpointGroupKey& group_key, in EndpointGroupExistsForTesting()
225 void SetEndpointForTesting(const ReportingEndpointGroupKey& group_key, in SetEndpointForTesting()
233 void SetV1EndpointForTesting(const ReportingEndpointGroupKey& group_key, in SetV1EndpointForTesting()
278 ReportingEndpointGroupKey group_key(kGroupKey); in SetEndpoint() local
Dreporting_endpoint.h132 ReportingEndpointGroupKey group_key; member
154 ReportingEndpointGroupKey group_key; member
179 ReportingEndpointGroupKey group_key; member
Dreporting_garbage_collector_unittest.cc94 ReportingEndpointGroupKey group_key(kNik_, kReportingSource_, in TEST_F() local
114 ReportingEndpointGroupKey group_key(kNik_, kReportingSource_, in TEST_F() local
Dmock_persistent_reporting_store.cc26 const ReportingEndpointGroupKey& group_key, in Command()
41 const ReportingEndpointGroupKey& group_key) in Command()
Dmock_persistent_reporting_store.h69 ReportingEndpointGroupKey group_key = ReportingEndpointGroupKey(); member
Dreporting_endpoint_manager.cc59 const ReportingEndpointGroupKey& group_key) override { in FindEndpointForDelivery()
Dreporting_endpoint.cc130 const ReportingEndpointGroupKey& group_key, in CachedReportingEndpointGroup()
Dreporting_cache_unittest.cc1567 ReportingEndpointGroupKey group_key(kNak_, kOrigin1_, in TEST_P() local
1591 ReportingEndpointGroupKey group_key(kNak_, kOrigin1_, in TEST_P() local
1668 ReportingEndpointGroupKey group_key(kNak_, url::Origin::Create(MakeURL(i)), in TEST_P() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dcollective_ops_test.py47 def all_reduce(t, group_size, group_key, instance_key, *args, **kwargs): argument
53 def all_gather(t, group_size, group_key, instance_key, *args, **kwargs): argument
65 def all_reduce(t, group_size, group_key, instance_key, *args, **kwargs): argument
73 def all_gather(t, group_size, group_key, instance_key, *args, **kwargs): argument
81 def broadcast_send(t, shape, dtype, group_size, group_key, instance_key, argument
90 def broadcast_recv(shape, dtype, group_size, group_key, instance_key, *args, argument
452 def run_all_reduce(group_size, group_key): argument
470 def run_and_assert(group_size, group_key): argument
/external/tensorflow/tensorflow/core/kernels/
Dcollective_ops.cc41 static string CollectiveKey(OpKernelContext* ctx, int32_t group_key, in CollectiveKey()
508 Tensor* group_key = nullptr; in Compute() local
528 Tensor* group_key) { in ComputeGroupKey()
590 const Tensor& group_size, const Tensor& group_key, in FillCollectiveParams()
910 CollectiveGroupResource(int32 group_key, int32 rank, int32 group_size, in CollectiveGroupResource()
928 int32 group_key() const { return group_key_; } in group_key() function in tensorflow::__anon8b310e240111::CollectiveGroupResource
983 auto group_key = group_key_t.unaligned_flat<int32>()(0); in ComputeAsync() local
/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()
Dtest_collective_executor_mgr.h63 Status LookupGroup(int32_t group_key, CollGroupParams* group) override { in LookupGroup()
Dcollective_param_resolver_local.cc502 void CollectiveParamResolverLocal::CancelGroup(int32 group_key) { in CancelGroup()
636 Status CollectiveParamResolverLocal::LookupGroup(int32_t group_key, in LookupGroup()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dall_reduce_op.cc39 int64_t group_key, group_size; in Compile() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_collective.cc77 LogicalResult SetCollectiveInfo(IntegerAttr group_size, IntegerAttr group_key, in SetCollectiveInfo()
101 IntegerAttr group_key = builder.getI32IntegerAttr(0); in SetCollectiveInfo() local
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto547 int32 group_key = 1; field
558 int32 group_key = 1; field
577 int32 group_key = 5; field
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed.cc206 CollectiveParamResolverDistributed::GetCachedGroup(int32_t group_key) { in GetCachedGroup()
317 int32_t group_key, int32_t instance_key) { in InstanceIsCached()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.h165 int group_key; member
Dxla_context.cc179 StatusOr<int64_t> XlaContext::RecordCollectiveInfo(int group_key, in RecordCollectiveInfo()
/external/tensorflow/tensorflow/core/framework/
Dcollective.h75 int32 group_key; member
340 virtual Status LookupGroup(int32_t group_key, CollGroupParams* group) { in LookupGroup()

12