Searched refs:deployment_mode (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/ |
| D | multi_process_cluster.py | 77 deployment_mode=data_service_pb2.DEPLOYMENT_MODE_COLOCATED): argument
79 self._deployment_mode = deployment_mode
94 deployment_mode=self._deployment_mode),
|
| D | auto_shard_test.py | 37 deployment_mode=data_service_pb2.DEPLOYMENT_MODE_COLOCATED): argument
45 deployment_mode=deployment_mode)
182 num_workers=5, local_shard_index=1, deployment_mode=None)
311 num_workers=5, local_shard_index=3, deployment_mode=None)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | metrics.cc | 370 int64_t worker_uid, tensorflow::data::DeploymentMode deployment_mode, in RecordTFDataServiceClientIterators() argument
374 tensorflow::data::DeploymentMode_Name(deployment_mode); in RecordTFDataServiceClientIterators()
|
| D | metrics.h | 122 int64_t worker_uid, tensorflow::data::DeploymentMode deployment_mode,
|
| /external/tensorflow/tensorflow/core/data/service/ |
| D | dispatcher_client_test.cc | 108 EXPECT_EQ(config.deployment_mode(), DEPLOYMENT_MODE_COLOCATED); in TEST_F()
|
| D | dispatcher.proto | 205 DeploymentMode deployment_mode = 4; field
|
| D | dispatcher_impl.cc | 571 response->mutable_config()->set_deployment_mode(config_.deployment_mode()); in GetDataServiceConfig()
1020 response->set_deployment_mode(config_.deployment_mode()); in ClientHeartbeat()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | data_service.proto | 91 DeploymentMode deployment_mode = 1; field
|
| D | service_config.proto | 30 DeploymentMode deployment_mode = 9; field
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | data_service_dataset_op.cc | 864 task.worker_uid(), resp.deployment_mode(), in RecordTFMetrics()
1377 config->deployment_mode() == DEPLOYMENT_MODE_COLOCATED && in MakeDataset()
|