/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
H A D | collective_ops.py | 21 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 D | reporting_cache_impl.cc | 244 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 D | reporting_endpoint_manager_unittest.cc | 52 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 D | reporting_test_util.cc | 217 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 D | reporting_header_parser.cc | 73 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 D | reporting_endpoint.h | 132 ReportingEndpointGroupKey group_key; member 154 ReportingEndpointGroupKey group_key; member 179 ReportingEndpointGroupKey group_key; member
|
H A D | reporting_garbage_collector_unittest.cc | 95 ReportingEndpointGroupKey group_key(kNak_, kReportingSource_, in TEST_F() local 115 ReportingEndpointGroupKey group_key(kNak_, kReportingSource_, in TEST_F() local
|
H A D | mock_persistent_reporting_store.cc | 26 const ReportingEndpointGroupKey& group_key, in Command() 41 const ReportingEndpointGroupKey& group_key) in Command()
|
H A D | reporting_cache_unittest.cc | 1572 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 D | mock_persistent_reporting_store.h | 69 ReportingEndpointGroupKey group_key = ReportingEndpointGroupKey(); member
|
H A D | reporting_endpoint_manager.cc | 59 const ReportingEndpointGroupKey& group_key) override { in FindEndpointForDelivery()
|
H A D | reporting_endpoint.cc | 131 const ReportingEndpointGroupKey& group_key, in CachedReportingEndpointGroup()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/ |
H A D | collective_ops_test.py | 47 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 D | collective_ops.cc | 41 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 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()
|
H A D | collective_param_resolver_local.cc | 502 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 D | worker.proto | 547 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 D | legalize_tf_collective.cc | 77 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 D | collective_param_resolver_distributed.cc | 206 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 D | collective.h | 75 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/ |
D | notification.pb.h | 657 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/ |
D | notification.pb.h | 657 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/ |
D | notification.pb.h | 686 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 D | notification.proto | 36 optional string group_key = 6 [ (.android.privacy).dest = DEST_EXPLICIT ]; field
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | all_reduce_op.cc | 39 int64_t group_key, group_size; in Compile() local
|