Home
last modified time | relevance | path

Searched refs:GrpcRPCFactory (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_rpc_factory.cc80 GrpcRPCFactory::GrpcRPCFactory(OpKernelConstruction* ctx, bool fail_fast, in GrpcRPCFactory() function in tensorflow::GrpcRPCFactory
96 GrpcRPCFactory::~GrpcRPCFactory() { in ~GrpcRPCFactory()
110 void GrpcRPCFactory::Call(OpKernelContext* ctx, int64 num_elements, in Call()
142 ::grpc::GenericStub* GrpcRPCFactory::GetOrCreateStubForAddress( in GetOrCreateStubForAddress()
155 GrpcRPCFactory::ChannelPtr GrpcRPCFactory::CreateChannelForAddress( in CreateChannelForAddress()
167 void GrpcRPCFactory::CreateCall(const Tensor& request_t, const bool try_rpc, in CreateCall()
188 void GrpcRPCFactory::StartCall(const Tensor& address_t, const Tensor& method_t, in StartCall()
Dgrpc_rpc_factory.h33 class GrpcRPCFactory : public RPCFactory {
35 explicit GrpcRPCFactory(OpKernelConstruction* ctx, bool fail_fast,
39 ~GrpcRPCFactory() override;
Dgrpc_rpc_factory_registration.cc27 return new GrpcRPCFactory(ctx, fail_fast, timeout_in_ms); in Function()