| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_rpc_factory.cc | 36 explicit GrpcCall(CallContainer<GrpcCall>* container, int index, bool try_rpc, in GrpcCall() argument
41 try_rpc_(try_rpc), in GrpcCall()
112 const Tensor& request_t, const bool try_rpc, in Call() argument
116 if (try_rpc) { in Call()
125 [this, &request_t, &try_rpc, response_t, status_code_t, status_message_t]( in Call()
127 CreateCall(request_t, try_rpc, index, container, response_t, in Call()
137 new CallContainer<GrpcCall>(ctx, num_elements, fail_fast_, try_rpc, in Call()
167 void GrpcRPCFactory::CreateCall(const Tensor& request_t, const bool try_rpc, in CreateCall() argument
178 if (try_rpc) { in CreateCall()
182 container->RegisterCall(container, index, try_rpc, get_request_ptr(index), in CreateCall()
[all …]
|
| D | grpc_rpc_factory.h | 42 const Tensor& method_t, const Tensor& request_t, const bool try_rpc,
53 void CreateCall(const Tensor& request_t, const bool try_rpc, int index,
|
| /external/tensorflow/tensorflow/contrib/rpc/python/kernel_tests/ |
| D | rpc_op_test_base.py | 48 def try_rpc(self, *args, **kwargs): member in RpcOpTestBase
49 return rpc_op.try_rpc(*args, protocol=self._protocol, **kwargs)
69 response_tensors, status_code, status_message = self.try_rpc(
106 self.try_rpc(method=method, address=self._address, request=''))
122 self.try_rpc(
140 response_tensors, status_code, status_message = self.try_rpc(
164 response_tensors, status_code, status_message = self.try_rpc(
271 response_tensors, status_code, status_message = self.try_rpc(
291 response_tensors, status_code, _ = self.try_rpc(
311 response_tensors, status_code, _ = self.try_rpc(
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | rpc_op.cc | 95 const bool try_rpc = (ctx->num_outputs() > 1); in ComputeAsync() local
99 if (try_rpc) { in ComputeAsync()
115 try_rpc, response_t, status_code_t, status_message_t, in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/util/rpc/ |
| D | call_container.h | 68 bool try_rpc, AsyncOpKernel::DoneCallback done,
99 OpKernelContext* ctx, int num_calls, bool fail_fast, bool try_rpc, in CallContainer() argument
109 try_rpc_(try_rpc), in CallContainer()
|
| D | rpc_factory.h | 60 const Tensor& request_t, const bool try_rpc,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | rpc_ops.cc | 24 Status RpcShapeOp(InferenceContext* c, bool try_rpc) { in RpcShapeOp() argument
45 if (try_rpc) { in RpcShapeOp()
|
| /external/tensorflow/tensorflow/contrib/rpc/ |
| D | __init__.py | 25 from tensorflow.contrib.rpc.python.ops.rpc_op import try_rpc
|
| /external/tensorflow/tensorflow/contrib/rpc/python/ops/ |
| D | rpc_op.py | 23 from tensorflow.contrib.rpc.python.ops.gen_rpc_op import try_rpc
|