| /aosp_15_r20/external/tensorflow/tensorflow/core/data/service/ |
| H A D | common_test.cc | 38 std::vector<ProcessingModeDef::ShardingPolicy> EnumerateShardingPolicies() { in EnumerateShardingPolicies()
39 std::vector<ProcessingModeDef::ShardingPolicy> result; in EnumerateShardingPolicies()
42 ProcessingModeDef::ShardingPolicy>(); in EnumerateShardingPolicies()
44 result.push_back(static_cast<ProcessingModeDef::ShardingPolicy>( in EnumerateShardingPolicies()
51 ProcessingModeDef processing_mode; in TEST()
52 processing_mode.set_sharding_policy(ProcessingModeDef::OFF); in TEST()
59 ProcessingModeDef processing_mode; in TEST()
60 processing_mode.set_sharding_policy(ProcessingModeDef::DYNAMIC); in TEST()
67 ProcessingModeDef processing_mode; in TEST()
68 std::vector<ProcessingModeDef::ShardingPolicy> policies = { in TEST()
[all …]
|
| H A D | common.cc | 40 bool IsNoShard(const ProcessingModeDef& processing_mode) { in IsNoShard()
41 return processing_mode.sharding_policy() == ProcessingModeDef::OFF; in IsNoShard()
44 bool IsDynamicShard(const ProcessingModeDef& processing_mode) { in IsDynamicShard()
45 return processing_mode.sharding_policy() == ProcessingModeDef::DYNAMIC; in IsDynamicShard()
48 bool IsStaticShard(const ProcessingModeDef& processing_mode) { in IsStaticShard()
49 return processing_mode.sharding_policy() == ProcessingModeDef::FILE || in IsStaticShard()
50 processing_mode.sharding_policy() == ProcessingModeDef::DATA || in IsStaticShard()
51 processing_mode.sharding_policy() == ProcessingModeDef::FILE_OR_DATA || in IsStaticShard()
52 processing_mode.sharding_policy() == ProcessingModeDef::HINT; in IsStaticShard()
55 Status ValidateProcessingMode(const ProcessingModeDef& processing_mode) { in ValidateProcessingMode()
[all …]
|
| H A D | data_service_test.cc | 76 dataset_client.Read(RangeDataset(20), ProcessingModeDef::OFF, in TEST()
93 dataset_client.Read(RangeDataset(20), ProcessingModeDef::DYNAMIC, in TEST()
105 ::testing::TestWithParam<ProcessingModeDef::ShardingPolicy>;
123 Values(ProcessingModeDef::FILE_OR_DATA,
124 ProcessingModeDef::DATA));
133 ProcessingModeDef::HINT, TARGET_WORKERS_LOCAL), in TEST()
157 dataset_client.Read(dataset, ProcessingModeDef::OFF, TARGET_WORKERS_ANY), in TEST()
178 EXPECT_THAT(dataset_client.Read(dataset, ProcessingModeDef::DATA, in TEST()
189 ::testing::TestWithParam<ProcessingModeDef::ShardingPolicy>;
204 dataset_client.Read(dataset, ProcessingModeDef::FILE_OR_DATA, in TEST_P()
[all …]
|
| H A D | dispatcher_client_test.cc | 154 ProcessingModeDef processing_mode; in TEST_F()
155 processing_mode.set_sharding_policy(ProcessingModeDef::OFF); in TEST_F()
181 ProcessingModeDef processing_mode; in TEST_F()
182 processing_mode.set_sharding_policy(ProcessingModeDef::OFF); in TEST_F()
206 ProcessingModeDef processing_mode; in TEST_F()
207 processing_mode.set_sharding_policy(ProcessingModeDef::OFF); in TEST_F()
215 processing_mode.set_sharding_policy(ProcessingModeDef::DYNAMIC); in TEST_F()
|
| H A D | common.h | 41 bool IsNoShard(const ProcessingModeDef& processing_mode);
44 bool IsDynamicShard(const ProcessingModeDef& processing_mode);
47 bool IsStaticShard(const ProcessingModeDef& processing_mode);
50 Status ValidateProcessingMode(const ProcessingModeDef& processing_mode);
55 ProcessingModeDef::ShardingPolicy sharding_policy);
|
| H A D | auto_shard_rewriter_test.cc | 72 TaskDef GetTaskDef(const ProcessingModeDef::ShardingPolicy sharding_policy, in GetTaskDef()
82 TaskDef task_def = GetTaskDef(ProcessingModeDef::FILE_OR_DATA, in TEST()
98 TaskDef task_def = GetTaskDef(ProcessingModeDef::DATA, /*num_workers=*/3, in TEST()
114 TaskDef task_def = GetTaskDef(ProcessingModeDef::FILE, /*num_workers=*/3, in TEST()
126 TaskDef task_def = GetTaskDef(ProcessingModeDef::HINT, /*num_workers=*/3, in TEST()
143 GetTaskDef(ProcessingModeDef::OFF, /*num_workers=*/3, /*worker_index=*/1); in TEST()
154 GetTaskDef(ProcessingModeDef::FILE_OR_DATA, /*num_workers=*/3, in TEST()
171 GetTaskDef(ProcessingModeDef::FILE_OR_DATA, /*num_workers=*/0, in TEST()
184 GetTaskDef(ProcessingModeDef::OFF, /*num_workers=*/0, /*worker_index=*/0); in TEST()
195 GetTaskDef(ProcessingModeDef::FILE_OR_DATA, /*num_workers=*/2, in TEST()
|
| H A D | test_cluster.h | 125 ProcessingModeDef::ShardingPolicy sharding_policy,
139 ProcessingModeDef::ShardingPolicy sharding_policy,
169 ProcessingModeDef::ShardingPolicy sharding_policy, in Read()
193 ProcessingModeDef::ShardingPolicy sharding_policy, in CreateIteration()
195 ProcessingModeDef processing_mode_def; in CreateIteration()
211 return CreateIteration(dataset_id, ProcessingModeDef::OFF, in CreateIteration()
|
| H A D | worker_client_test.cc | 78 ProcessingModeDef processing_mode; in CreateIteration()
79 processing_mode.set_sharding_policy(ProcessingModeDef::OFF); in CreateIteration()
|
| H A D | dispatcher_state.h | 155 const ProcessingModeDef& processing_mode, std::string job_name, in Job()
168 const ProcessingModeDef processing_mode;
|
| H A D | export.proto | 16 ProcessingModeDef processing_mode = 4;
|
| H A D | dispatcher_client.h | 79 const ProcessingModeDef& processing_mode,
|
| H A D | common.proto | 38 ProcessingModeDef processing_mode_def = 10;
|
| H A D | journal.proto | 52 ProcessingModeDef processing_mode_def = 4;
|
| H A D | dispatcher_client.cc | 140 const std::string& dataset_id, const ProcessingModeDef& processing_mode, in GetOrCreateJob()
|
| H A D | dispatcher.proto | 117 ProcessingModeDef processing_mode_def = 2;
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/ |
| D | data_service.pb.cc | 23 PROTOBUF_CONSTEXPR ProcessingModeDef::ProcessingModeDef( in ProcessingModeDef() function in tensorflow::data::ProcessingModeDef
32 ProcessingModeDef _instance;
150 constexpr ProcessingModeDef_ShardingPolicy ProcessingModeDef::OFF;
151 constexpr ProcessingModeDef_ShardingPolicy ProcessingModeDef::DYNAMIC;
152 constexpr ProcessingModeDef_ShardingPolicy ProcessingModeDef::FILE;
153 constexpr ProcessingModeDef_ShardingPolicy ProcessingModeDef::DATA;
154 constexpr ProcessingModeDef_ShardingPolicy ProcessingModeDef::FILE_OR_DATA;
155 constexpr ProcessingModeDef_ShardingPolicy ProcessingModeDef::HINT;
156 constexpr ProcessingModeDef_ShardingPolicy ProcessingModeDef::ShardingPolicy_MIN;
157 constexpr ProcessingModeDef_ShardingPolicy ProcessingModeDef::ShardingPolicy_MAX;
[all …]
|
| D | data_service.pb.h | 57 class ProcessingModeDef; variable
66 template<> ::tensorflow::data::ProcessingModeDef* Arena::CreateMaybeMessage<::tensorflow::data::Pro…
143 class ProcessingModeDef final :
146 inline ProcessingModeDef() : ProcessingModeDef(nullptr) {} in ProcessingModeDef() function
147 ~ProcessingModeDef() override;
148 …explicit PROTOBUF_CONSTEXPR ProcessingModeDef(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitializ…
150 ProcessingModeDef(const ProcessingModeDef& from);
151 ProcessingModeDef(ProcessingModeDef&& from) noexcept in ProcessingModeDef() function
152 : ProcessingModeDef() { in ProcessingModeDef()
156 inline ProcessingModeDef& operator=(const ProcessingModeDef& from) {
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
| H A D | data_service_ops.py | 110 return data_service_pb2.ProcessingModeDef.OFF
112 return data_service_pb2.ProcessingModeDef.DYNAMIC
114 return data_service_pb2.ProcessingModeDef.FILE
116 return data_service_pb2.ProcessingModeDef.DATA
118 return data_service_pb2.ProcessingModeDef.FILE_OR_DATA
120 return data_service_pb2.ProcessingModeDef.HINT
314 processing_mode_def = data_service_pb2.ProcessingModeDef(
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | metrics.cc | 357 const tensorflow::data::ProcessingModeDef& processing_mode, in RecordTFDataServiceJobsCreated()
360 data::ProcessingModeDef::ShardingPolicy_Name( in RecordTFDataServiceJobsCreated()
371 const tensorflow::data::ProcessingModeDef& processing_mode, in RecordTFDataServiceClientIterators()
376 data::ProcessingModeDef::ShardingPolicy_Name( in RecordTFDataServiceClientIterators()
|
| H A D | metrics.h | 117 const tensorflow::data::ProcessingModeDef& processing_mode,
123 const tensorflow::data::ProcessingModeDef& processing_mode,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | data_service_dataset_op.cc | 182 const ProcessingModeDef& processing_mode, const std::string& address, in Dataset()
273 if (processing_mode_.sharding_policy() == ProcessingModeDef::OFF || in CardinalityInternal()
274 processing_mode_.sharding_policy() == ProcessingModeDef::DYNAMIC || in CardinalityInternal()
275 processing_mode_.sharding_policy() == ProcessingModeDef::DATA) { in CardinalityInternal()
1257 const ProcessingModeDef processing_mode_;
1348 ProcessingModeDef processing_mode; in MakeDataset()
1350 processing_mode.set_sharding_policy(ProcessingModeDef::OFF); in MakeDataset()
1352 processing_mode.set_sharding_policy(ProcessingModeDef::DYNAMIC); in MakeDataset()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/service/client/ |
| H A D | validate_utils_test.cc | 41 params.processing_mode.set_sharding_policy(ProcessingModeDef::OFF); in GetDefaultParams()
83 params.processing_mode.set_sharding_policy(ProcessingModeDef::FILE_OR_DATA); in TEST()
|
| H A D | common.h | 33 ProcessingModeDef processing_mode;
|
| H A D | validate_utils.cc | 39 ProcessingModeDef::ShardingPolicy_Name( in ValidateLocalWorkers()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/ |
| H A D | data_service.proto | 8 message ProcessingModeDef { message
|