Searched refs:TensorResponse (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | tensor_coding.cc | 26 TensorResponse::Source::~Source() {} in ~Source()
28 void TensorResponse::Clear() { in Clear()
37 void TensorResponse::ClearTensor() { in ClearTensor()
42 void TensorResponse::InitAlloc(DeviceBase* d, const AllocatorAttributes& aa) { in InitAlloc()
53 Status TensorResponse::InitFrom(RecvTensorResponse* response) { in InitFrom()
71 void TensorResponse::InitPartial(const RecvTensorResponse& response, in InitPartial()
82 Status TensorResponse::ParseFrom(Source* source) { in ParseFrom()
147 bool TensorResponse::ParseTensorSubmessage( in ParseTensorSubmessage()
219 bool TensorResponse::ParseFast(Source* source) { in ParseFast()
280 bool TensorResponse::ParseSlow(Source* source) { in ParseSlow()
|
| D | tensor_coding.h | 35 class TensorResponse {
37 TensorResponse() {} in TensorResponse() function
|
| D | tensor_coding_test.cc | 49 class StringSource : public TensorResponse::Source {
91 TensorResponse response; in Validate()
183 TensorResponse response; in BM_TensorResponse()
|
| D | worker_interface.h | 34 class TensorResponse; variable
106 TensorResponse* response,
|
| D | worker.h | 86 TensorResponse* response, StatusCallback done) override;
|
| D | test_utils.h | 83 TensorResponse* response, StatusCallback done) override { in RecvTensorAsync()
|
| D | worker.cc | 458 TensorResponse* response, StatusCallback done) { in RecvTensorAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service_impl.h | 40 class SerializationTraits<tensorflow::TensorResponse> {
42 static Status Serialize(const tensorflow::TensorResponse& msg, ByteBuffer* bp, in Serialize()
48 tensorflow::TensorResponse* msg) { in Deserialize()
|
| D | grpc_util.h | 43 class GrpcByteSource : public TensorResponse::Source {
125 bool GrpcMaybeParseProto(::grpc::ByteBuffer* src, TensorResponse* dst);
|
| D | grpc_remote_worker.cc | 198 TensorResponse* response, StatusCallback done) override { in RecvTensorAsync()
280 void IssueRequest(const protobuf::Message* request, TensorResponse* response, in IssueRequest()
283 new RPCState<TensorResponse>(&stub_, cq_, method, *request, response, in IssueRequest()
|
| D | grpc_util.cc | 97 bool GrpcMaybeParseProto(::grpc::ByteBuffer* src, TensorResponse* dst) { in GrpcMaybeParseProto()
|
| D | rpc_rendezvous_mgr.cc | 178 TensorResponse resp_;
|
| D | rpc_rendezvous_mgr_test.cc | 59 TensorResponse* response, StatusCallback done) override { in RecvTensorAsync()
|
| D | BUILD | 49 # Required to be able to overload TensorResponse parsing.
|