Home
last modified time | relevance | path

Searched refs:address_t (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Drpc_op.cc62 const Tensor& address_t = ctx->input(0); in ComputeAsync() local
67 ctx, address_t.dims() == 0 || address_t.dims() == 1, in ComputeAsync()
77 for (const Tensor& t : {address_t, method_t, request_t}) { in ComputeAsync()
114 rpc_factory_->Call(ctx, num_elements, address_t, method_t, request_t, in ComputeAsync()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_rpc_factory.cc111 const Tensor& address_t, const Tensor& method_t, in Call() argument
132 [this, &address_t, &method_t](GrpcCall* call) { in Call()
133 StartCall(address_t, method_t, call); in Call()
188 void GrpcRPCFactory::StartCall(const Tensor& address_t, const Tensor& method_t, in StartCall() argument
190 auto address = address_t.flat<string>(); in StartCall()
Dgrpc_rpc_factory.h41 void Call(OpKernelContext* ctx, int64 num_elements, const Tensor& address_t,
60 void StartCall(const Tensor& address_t, const Tensor& method_t,
/external/tensorflow/tensorflow/core/util/rpc/
Drpc_factory.h59 const Tensor& address_t, const Tensor& method_t,