1 #ifndef TENSORFLOW_COMPILER_XLA_PYTHON_TPU_DRIVER_GRPC_TPU_DRIVER_H_ 2 #define TENSORFLOW_COMPILER_XLA_PYTHON_TPU_DRIVER_GRPC_TPU_DRIVER_H_ 3 4 // Copyright 2019 The TensorFlow Authors. All Rights Reserved. 5 // 6 // Licensed under the Apache License, Version 2.0 (the "License"); 7 // you may not use this file except in compliance with the License. 8 // You may obtain a copy of the License at 9 // 10 // http://www.apache.org/licenses/LICENSE-2.0 11 // 12 // Unless required by applicable law or agreed to in writing, software 13 // distributed under the License is distributed on an "AS IS" BASIS, 14 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15 // See the License for the specific language governing permissions and 16 // limitations under the License. 17 // ============================================================================== 18 19 #include "grpcpp/grpcpp.h" 20 #include "tensorflow/compiler/xla/python/tpu_driver/tpu_driver.h" 21 #include "tensorflow/compiler/xla/python/tpu_driver/tpu_driver.pb.h" 22 23 namespace tpu_driver { 24 25 xla::StatusOr<std::unique_ptr<TpuDriver>> CreateGrpcTpuDriver( 26 const TpuDriverConfig& config, 27 std::shared_ptr<grpc::ChannelCredentials> credentials); 28 29 } // namespace tpu_driver 30 31 #endif // TENSORFLOW_COMPILER_XLA_PYTHON_TPU_DRIVER_GRPC_TPU_DRIVER_H_ 32