Searched refs:serving_port (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_pod_state.cc | 54 ConstructCacheService(ResourceMgr* rmgr, int serving_port, in ConstructCacheService() argument
58 server_builder = tpu::CreateServerBuilder(serving_port); in ConstructCacheService()
60 server_builder = tpu::CreateServerBuilderGoogle(serving_port); in ConstructCacheService()
72 Status GetServerAddressAndPort(std::string* server_address, int* serving_port) { in GetServerAddressAndPort() argument
80 *serving_port = -1; in GetServerAddressAndPort()
87 params.port_output = serving_port; in GetServerAddressAndPort()
94 CHECK_NE(*serving_port, -1); in GetServerAddressAndPort()
153 int serving_port; in ConstructTpuPodState() local
155 TF_RETURN_IF_ERROR(GetServerAddressAndPort(&server_address, &serving_port)); in ConstructTpuPodState()
180 ConstructCacheService(rmgr, serving_port, compilation_cache)); in ConstructTpuPodState()
[all …]
|
| D | tpu_util.cc | 102 int serving_port) { in CreateServerBuilder() argument
105 absl::StrFormat("[::]:%d", serving_port), in CreateServerBuilder()
|
| D | tpu_pod_state.h | 60 Status GetServerAddressAndPort(std::string* server_address, int* serving_port);
|
| D | tpu_util.h | 62 int serving_port);
|