Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A 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
/aosp_15_r20/external/cronet/net/reporting/
H A Dreporting_cache_impl.cc244 const ReportingEndpointGroupKey& group_key, in GetEndpointStats()
273 const ReportingEndpointGroupKey& group_key, in IncrementEndpointDeliveries()
534 const ReportingEndpointGroupKey& group_key) { in RemoveEndpointGroup()
570 const ReportingEndpointGroupKey& group_key = endpoint_it->first; in RemoveEndpointsForUrl() local
615 const ReportingEndpointGroupKey& group_key = group.group_key; in AddClientsLoadedFromStore() local
701 const ReportingEndpointGroupKey& group_key) { in GetCandidateEndpointsForDelivery()
813 const ReportingEndpointGroupKey& group_key, in GetEndpointForTesting()
825 const ReportingEndpointGroupKey& group_key, in EndpointGroupExistsForTesting()
869 const ReportingEndpointGroupKey& group_key, in SetV1EndpointForTesting()
906 const ReportingEndpointGroupKey& group_key, in SetEndpointForTesting()
[all …]
H A Dreporting_endpoint_manager_unittest.cc52 const ReportingEndpointGroupKey& group_key) override { in GetCandidateEndpointsForDelivery()
106 void IncrementEndpointDeliveries(const ReportingEndpointGroupKey& group_key, in IncrementEndpointDeliveries()
175 const ReportingEndpointGroupKey& group_key) override { in RemoveEndpointGroup()
205 const ReportingEndpointGroupKey& group_key, in GetEndpointForTesting()
210 bool EndpointGroupExistsForTesting(const ReportingEndpointGroupKey& group_key, in EndpointGroupExistsForTesting()
234 void SetEndpointForTesting(const ReportingEndpointGroupKey& group_key, in SetEndpointForTesting()
242 void SetV1EndpointForTesting(const ReportingEndpointGroupKey& group_key, in SetV1EndpointForTesting()
287 ReportingEndpointGroupKey group_key(kGroupKey); in SetEndpoint() local
H A Dreporting_test_util.cc217 const ReportingEndpointGroupKey& group_key, in FindEndpointInCache()
223 const ReportingEndpointGroupKey& group_key, in SetEndpointInCache()
236 const ReportingEndpointGroupKey& group_key, in SetV1EndpointInCache()
245 const ReportingEndpointGroupKey& group_key, in EndpointExistsInCache()
252 const ReportingEndpointGroupKey& group_key, in GetEndpointStatistics()
267 const ReportingEndpointGroupKey& group_key, in EndpointGroupExistsInCache()
H A 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
H A Dreporting_endpoint.h132 ReportingEndpointGroupKey group_key; member
154 ReportingEndpointGroupKey group_key; member
179 ReportingEndpointGroupKey group_key; member
H A Dreporting_garbage_collector_unittest.cc95 ReportingEndpointGroupKey group_key(kNak_, kReportingSource_, in TEST_F() local
115 ReportingEndpointGroupKey group_key(kNak_, kReportingSource_, in TEST_F() local
H A Dmock_persistent_reporting_store.cc26 const ReportingEndpointGroupKey& group_key, in Command()
41 const ReportingEndpointGroupKey& group_key) in Command()
H A Dreporting_cache_unittest.cc1572 ReportingEndpointGroupKey group_key(kNak_, kOrigin1_, in TEST_P() local
1596 ReportingEndpointGroupKey group_key(kNak_, kOrigin1_, in TEST_P() local
1664 ReportingEndpointGroupKey group_key(kNak_, url::Origin::Create(MakeURL(i)), in TEST_P() local
H A Dmock_persistent_reporting_store.h69 ReportingEndpointGroupKey group_key = ReportingEndpointGroupKey(); member
H A Dreporting_endpoint_manager.cc59 const ReportingEndpointGroupKey& group_key) override { in FindEndpointForDelivery()
H A Dreporting_endpoint.cc131 const ReportingEndpointGroupKey& group_key, in CachedReportingEndpointGroup()
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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::__anon3faef4750111::CollectiveGroupResource
983 auto group_key = group_key_t.unaligned_flat<int32>()(0); in ComputeAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A 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()
H A Dcollective_param_resolver_local.cc502 void CollectiveParamResolverLocal::CancelGroup(int32 group_key) { in CancelGroup()
636 Status CollectiveParamResolverLocal::LookupGroup(int32_t group_key, in LookupGroup()
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dworker.proto547 int32 group_key = 1; field
558 int32 group_key = 1; field
577 int32 group_key = 5; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf_collective.cc77 LogicalResult SetCollectiveInfo(IntegerAttr group_size, IntegerAttr group_key, in SetCollectiveInfo()
101 IntegerAttr group_key = builder.getI32IntegerAttr(0); in SetCollectiveInfo() local
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dcollective_param_resolver_distributed.cc206 CollectiveParamResolverDistributed::GetCachedGroup(int32_t group_key) { in GetCachedGroup()
317 int32_t group_key, int32_t instance_key) { in InstanceIsCached()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcollective.h75 int32 group_key; member
340 virtual Status LookupGroup(int32_t group_key, CollGroupParams* group) { in LookupGroup()
/aosp_15_r20/out/soong/.intermediates/frameworks/base/libplatformprotos/android_arm64_armv8-2a_cortex-a55_static_cfi/gen/proto/frameworks/base/core/proto/android/app/
Dnotification.pb.h657 inline const std::string& NotificationProto::group_key() const { in group_key() function
698 inline void NotificationProto::set_allocated_group_key(std::string* group_key) { in set_allocated_group_key()
/aosp_15_r20/out/soong/.intermediates/frameworks/base/libplatformprotos/android_arm64_armv8-2a_cortex-a55_static/gen/proto/frameworks/base/core/proto/android/app/
Dnotification.pb.h657 inline const std::string& NotificationProto::group_key() const { in group_key() function
698 inline void NotificationProto::set_allocated_group_key(std::string* group_key) { in set_allocated_group_key()
/aosp_15_r20/out/soong/.intermediates/frameworks/base/libplatformprotos/linux_glibc_x86_64_static/gen/proto/frameworks/base/core/proto/android/app/
Dnotification.pb.h686 inline const std::string& NotificationProto::group_key() const { in group_key() function
727 inline void NotificationProto::set_allocated_group_key(std::string* group_key) { in set_allocated_group_key()
/aosp_15_r20/frameworks/base/core/proto/android/app/
H A Dnotification.proto36 optional string group_key = 6 [ (.android.privacy).dest = DEST_EXPLICIT ]; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dall_reduce_op.cc39 int64_t group_key, group_size; in Compile() local

12