Searched defs:processing_mode (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/data/service/ |
| D | common.cc | 36 bool IsNoShard(const ProcessingModeDef& processing_mode) { in IsNoShard()
40 bool IsDynamicShard(const ProcessingModeDef& processing_mode) { in IsDynamicShard()
44 bool IsStaticShard(const ProcessingModeDef& processing_mode) { in IsStaticShard()
51 Status ValidateProcessingMode(const ProcessingModeDef& processing_mode) { in ValidateProcessingMode()
|
| 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
|
| D | worker_client_test.cc | 77 ProcessingModeDef processing_mode; in CreateJob() local
|
| D | dispatcher_client.cc | 150 int64_t dataset_id, const ProcessingModeDef& processing_mode, in GetOrCreateJob()
|
| D | dispatcher_state.h | 170 const ProcessingModeDef processing_mode; member
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | data_service_ops.py | 128 def _get_validated_sharding_policy(processing_mode): argument
147 def _serialize(processing_mode): argument
181 processing_mode, argument
312 def __init__(self, dataset_id, processing_mode, address, element_spec, argument
368 def _distribute(processing_mode, argument
461 def distribute(processing_mode, argument
804 def _from_dataset_id(processing_mode, argument
955 def from_dataset_id(processing_mode, argument
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/ |
| D | test_base.py | 217 processing_mode="parallel_epochs", argument
240 processing_mode="parallel_epochs", argument
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | data_service_dataset_op.cc | 103 const ProcessingModeDef& processing_mode, const std::string& address, in Dataset()
184 Node* processing_mode; in AsGraphDefInternal() local
1081 ProcessingModeDef processing_mode; in MakeDataset() local
|