Home
last modified time | relevance | path

Searched refs:GetDeviceHandles (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/rpc/
Dgrpc_service.cc50 ::grpc::Status GRPCService::GetDeviceHandles(::grpc::ServerContext* context, in GetDeviceHandles() function in xla::GRPCService
54 return service_->GetDeviceHandles(arg, result); in GetDeviceHandles()
Dgrpc_stub.cc116 Status GRPCStub::GetDeviceHandles(const GetDeviceHandlesRequest* request, in GetDeviceHandles() function in xla::GRPCStub
119 return grpc_stub_->GetDeviceHandles(context, *request, response); in GetDeviceHandles()
Dgrpc_stub.h67 Status GetDeviceHandles(const GetDeviceHandlesRequest* arg,
Dgrpc_service.h42 ::grpc::Status GetDeviceHandles(::grpc::ServerContext* context,
Dxla_service.proto120 rpc GetDeviceHandles(GetDeviceHandlesRequest) rpc
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc292 GetDeviceHandles(/*device_count=*/1)); in Execute()
365 StatusOr<std::vector<DeviceHandle>> Client::GetDeviceHandles( in GetDeviceHandles() function in xla::Client
375 Status s = stub_->GetDeviceHandles(&request, &response); in GetDeviceHandles()
Dclient.h126 StatusOr<std::vector<DeviceHandle>> GetDeviceHandles(int64 device_count);
/external/tensorflow/tensorflow/compiler/xla/service/
Dcompile_only_service.h62 Status GetDeviceHandles(const GetDeviceHandlesRequest* arg, in GetDeviceHandles() function
Dservice.h125 Status GetDeviceHandles(const GetDeviceHandlesRequest* arg,
Dservice.cc759 Status Service::GetDeviceHandles(const GetDeviceHandlesRequest* arg, in GetDeviceHandles() function in xla::Service
/external/tensorflow/tensorflow/compiler/xla/
Dservice_interface.h75 virtual Status GetDeviceHandles(const GetDeviceHandlesRequest* arg,
/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_test.cc133 client_->GetDeviceHandles(1)); in XLA_TEST_F()
Dgather_operation_test.cc662 client_->GetDeviceHandles(1)); in XLA_TEST_F()