Home
last modified time | relevance | path

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

123

/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
/external/cronet/stable/net/reporting/
Dreporting_cache_impl.cc253 const ReportingEndpointGroupKey& group_key, in GetEndpointStats()
282 const ReportingEndpointGroupKey& group_key, in IncrementEndpointDeliveries()
572 const ReportingEndpointGroupKey& group_key) { in RemoveEndpointGroup()
608 const ReportingEndpointGroupKey& group_key = endpoint_it->first; in RemoveEndpointsForUrl() local
653 const ReportingEndpointGroupKey& group_key = group.group_key; in AddClientsLoadedFromStore() local
741 const ReportingEndpointGroupKey& group_key) { in GetCandidateEndpointsForDelivery()
867 const ReportingEndpointGroupKey& group_key, in GetEndpointForTesting()
884 const ReportingEndpointGroupKey& group_key, in EndpointGroupExistsForTesting()
928 const ReportingEndpointGroupKey& group_key, in SetV1EndpointForTesting()
967 const ReportingEndpointGroupKey& group_key, in SetEnterpriseEndpointForTesting()
[all …]
Dreporting_test_util.cc223 const ReportingEndpointGroupKey& group_key, in FindEndpointInCache()
229 const ReportingEndpointGroupKey& group_key, in SetEndpointInCache()
242 const ReportingEndpointGroupKey& group_key, in SetV1EndpointInCache()
251 const ReportingEndpointGroupKey& group_key, in SetEnterpriseEndpointInCache()
257 const ReportingEndpointGroupKey& group_key, in EndpointExistsInCache()
264 const ReportingEndpointGroupKey& group_key, in GetEndpointStatistics()
279 const ReportingEndpointGroupKey& group_key, in EndpointGroupExistsInCache()
Dreporting_endpoint_manager_unittest.cc53 const ReportingEndpointGroupKey& group_key) override { in GetCandidateEndpointsForDelivery()
105 void IncrementEndpointDeliveries(const ReportingEndpointGroupKey& group_key, in IncrementEndpointDeliveries()
168 const ReportingEndpointGroupKey& group_key) override { in RemoveEndpointGroup()
191 const ReportingEndpointGroupKey& group_key, in GetEndpointForTesting()
199 bool EndpointGroupExistsForTesting(const ReportingEndpointGroupKey& group_key, in EndpointGroupExistsForTesting()
212 void SetEndpointForTesting(const ReportingEndpointGroupKey& group_key, in SetEndpointForTesting()
220 void SetV1EndpointForTesting(const ReportingEndpointGroupKey& group_key, in SetV1EndpointForTesting()
227 const ReportingEndpointGroupKey& group_key, in SetEnterpriseEndpointForTesting()
269 ReportingEndpointGroupKey group_key(kGroupKey); in SetEndpoint() local
282 ReportingEndpointGroupKey group_key(kEnterpriseGroupKey); in SetEnterpriseEndpoint() local
Dreporting_header_parser.cc74 const ReportingEndpointGroupKey& group_key, in ProcessEndpoint()
144 ReportingEndpointGroupKey group_key(network_anonymization_key, origin, in ProcessEndpointGroup() local
202 const ReportingEndpointGroupKey& group_key, in ProcessEndpoint()
238 ReportingEndpointGroupKey group_key(network_anonymization_key, in ProcessV1Endpoint() local
Dreporting_endpoint.h163 ReportingEndpointGroupKey group_key; member
185 ReportingEndpointGroupKey group_key; member
210 ReportingEndpointGroupKey group_key; member
Dreporting_garbage_collector_unittest.cc100 ReportingEndpointGroupKey group_key(kNak_, kReportingSource_, in TEST_F() local
121 ReportingEndpointGroupKey group_key(kNak_, 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.cc138 const ReportingEndpointGroupKey& group_key, in CachedReportingEndpointGroup()
/external/cronet/tot/net/reporting/
Dreporting_cache_impl.cc253 const ReportingEndpointGroupKey& group_key, in GetEndpointStats()
282 const ReportingEndpointGroupKey& group_key, in IncrementEndpointDeliveries()
572 const ReportingEndpointGroupKey& group_key) { in RemoveEndpointGroup()
608 const ReportingEndpointGroupKey& group_key = endpoint_it->first; in RemoveEndpointsForUrl() local
653 const ReportingEndpointGroupKey& group_key = group.group_key; in AddClientsLoadedFromStore() local
741 const ReportingEndpointGroupKey& group_key) { in GetCandidateEndpointsForDelivery()
867 const ReportingEndpointGroupKey& group_key, in GetEndpointForTesting()
884 const ReportingEndpointGroupKey& group_key, in EndpointGroupExistsForTesting()
928 const ReportingEndpointGroupKey& group_key, in SetV1EndpointForTesting()
967 const ReportingEndpointGroupKey& group_key, in SetEnterpriseEndpointForTesting()
[all …]
Dreporting_test_util.cc223 const ReportingEndpointGroupKey& group_key, in FindEndpointInCache()
229 const ReportingEndpointGroupKey& group_key, in SetEndpointInCache()
242 const ReportingEndpointGroupKey& group_key, in SetV1EndpointInCache()
251 const ReportingEndpointGroupKey& group_key, in SetEnterpriseEndpointInCache()
257 const ReportingEndpointGroupKey& group_key, in EndpointExistsInCache()
264 const ReportingEndpointGroupKey& group_key, in GetEndpointStatistics()
279 const ReportingEndpointGroupKey& group_key, in EndpointGroupExistsInCache()
Dreporting_endpoint_manager_unittest.cc53 const ReportingEndpointGroupKey& group_key) override { in GetCandidateEndpointsForDelivery()
105 void IncrementEndpointDeliveries(const ReportingEndpointGroupKey& group_key, in IncrementEndpointDeliveries()
168 const ReportingEndpointGroupKey& group_key) override { in RemoveEndpointGroup()
191 const ReportingEndpointGroupKey& group_key, in GetEndpointForTesting()
199 bool EndpointGroupExistsForTesting(const ReportingEndpointGroupKey& group_key, in EndpointGroupExistsForTesting()
212 void SetEndpointForTesting(const ReportingEndpointGroupKey& group_key, in SetEndpointForTesting()
220 void SetV1EndpointForTesting(const ReportingEndpointGroupKey& group_key, in SetV1EndpointForTesting()
227 const ReportingEndpointGroupKey& group_key, in SetEnterpriseEndpointForTesting()
269 ReportingEndpointGroupKey group_key(kGroupKey); in SetEndpoint() local
282 ReportingEndpointGroupKey group_key(kEnterpriseGroupKey); in SetEnterpriseEndpoint() local
Dreporting_header_parser.cc74 const ReportingEndpointGroupKey& group_key, in ProcessEndpoint()
144 ReportingEndpointGroupKey group_key(network_anonymization_key, origin, in ProcessEndpointGroup() local
202 const ReportingEndpointGroupKey& group_key, in ProcessEndpoint()
238 ReportingEndpointGroupKey group_key(network_anonymization_key, in ProcessV1Endpoint() local
Dreporting_endpoint.h163 ReportingEndpointGroupKey group_key; member
185 ReportingEndpointGroupKey group_key; member
210 ReportingEndpointGroupKey group_key; member
Dreporting_garbage_collector_unittest.cc100 ReportingEndpointGroupKey group_key(kNak_, kReportingSource_, in TEST_F() local
121 ReportingEndpointGroupKey group_key(kNak_, 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.cc138 const ReportingEndpointGroupKey& group_key, in CachedReportingEndpointGroup()
/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::__anon33201ef50111::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()
/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

123