Searched refs:CreateChannelHandle (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_client.h | 118 StatusOr<ChannelHandle> CreateChannelHandle() { in CreateChannelHandle() function
119 return pjrt_client_->CreateChannelHandle(); in CreateChannelHandle()
|
| D | xla.cc | 217 .def("create_channel_handle", &PyClient::CreateChannelHandle) in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/compiler/xla/rpc/ |
| D | grpc_stub.cc | 123 Status GRPCStub::CreateChannelHandle(const CreateChannelHandleRequest* request, in CreateChannelHandle() function in xla::GRPCStub
126 return grpc_stub_->CreateChannelHandle(context, *request, response); in CreateChannelHandle()
|
| D | grpc_stub.h | 70 Status CreateChannelHandle(const CreateChannelHandleRequest* arg,
|
| D | xla_service.proto | 126 rpc CreateChannelHandle(CreateChannelHandleRequest) rpc
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | service_interface.h | 72 virtual Status CreateChannelHandle(const CreateChannelHandleRequest* arg,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client.h | 218 StatusOr<ChannelHandle> CreateChannelHandle();
|
| D | client.cc | 494 Status s = stub_->CreateChannelHandle(&request, &response); in CreateChannelHandleByType()
504 StatusOr<ChannelHandle> Client::CreateChannelHandle() { in CreateChannelHandle() function in xla::Client
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | service.h | 179 Status CreateChannelHandle(const CreateChannelHandleRequest* arg,
|
| D | service.cc | 186 Status Service::CreateChannelHandle(const CreateChannelHandleRequest* arg, in CreateChannelHandle() function in xla::Service
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.h | 204 StatusOr<ChannelHandle> CreateChannelHandle() override { in CreateChannelHandle() function
205 return client()->CreateChannelHandle(); in CreateChannelHandle()
|
| D | pjrt_client.h | 262 virtual StatusOr<ChannelHandle> CreateChannelHandle() = 0;
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 1408 TF_ASSIGN_OR_RETURN(result.first->second, client()->CreateChannelHandle()); in GetChannelHandle()
|