| /aosp_15_r20/external/tensorflow/tensorflow/core/data/service/ |
| H A D | common_test.cc | 51 ProcessingModeDef processing_mode; in TEST() local
59 ProcessingModeDef processing_mode; in TEST() local
67 ProcessingModeDef processing_mode; in TEST() local
80 ProcessingModeDef processing_mode; in TEST() local
115 ProcessingModeDef processing_mode; in TEST() local
122 ProcessingModeDef processing_mode; in TEST() local
|
| H A D | common.cc | 40 bool IsNoShard(const ProcessingModeDef& processing_mode) { in IsNoShard()
44 bool IsDynamicShard(const ProcessingModeDef& processing_mode) { in IsDynamicShard()
48 bool IsStaticShard(const ProcessingModeDef& processing_mode) { in IsStaticShard()
55 Status ValidateProcessingMode(const ProcessingModeDef& processing_mode) { in ValidateProcessingMode()
|
| H A D | dispatcher_client_test.cc | 154 ProcessingModeDef processing_mode; in TEST_F() local
181 ProcessingModeDef processing_mode; in TEST_F() local
206 ProcessingModeDef processing_mode; in TEST_F() local
|
| H A D | export.proto | 16 ProcessingModeDef processing_mode = 4; field
|
| H A D | worker_client_test.cc | 78 ProcessingModeDef processing_mode; in CreateIteration() local
|
| H A D | dispatcher_client.cc | 140 const std::string& dataset_id, const ProcessingModeDef& processing_mode, in GetOrCreateJob()
|
| H A D | dispatcher_state.h | 168 const ProcessingModeDef processing_mode; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
| H A D | data_service_ops.py | 165 def _get_validated_sharding_policy(processing_mode): argument
241 processing_mode, argument
402 def __init__(self, dataset_id, processing_mode, address, element_spec, argument
459 def _distribute(processing_mode, argument
555 def distribute(processing_mode, argument
925 def _from_dataset_id(processing_mode, argument
1084 def from_dataset_id(processing_mode, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/ |
| H A D | test_base.py | 242 processing_mode="parallel_epochs", argument
269 processing_mode="parallel_epochs", argument
|
| /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()
315 Node* processing_mode; in AsGraphDefInternal() local
1348 ProcessingModeDef processing_mode; in MakeDataset() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | metrics.cc | 357 const tensorflow::data::ProcessingModeDef& processing_mode, in RecordTFDataServiceJobsCreated()
371 const tensorflow::data::ProcessingModeDef& processing_mode, in RecordTFDataServiceClientIterators()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/service/client/ |
| H A D | common.h | 33 ProcessingModeDef processing_mode; member
|
| /aosp_15_r20/external/libva/va/ |
| H A D | va_vpp.h | 1122 VAProcMode processing_mode; member
|