Searched refs:send_tensor (Results 1 – 5 of 5) sorted by relevance
1061 auto* send_tensor = remote_enqueue_request.add_queue()->mutable_send_tensor(); in TEST_F() local1062 send_tensor->set_op_id(1); in TEST_F()1063 SetTensorProto(send_tensor->add_tensors()); in TEST_F()1134 auto* send_tensor = remote_enqueue_request.add_queue()->mutable_send_tensor(); in TEST_F() local1135 send_tensor->set_op_id(1); in TEST_F()1136 SetTensorProto(send_tensor->add_tensors()); in TEST_F()1144 remote_handle->set_op_id(send_tensor->op_id()); in TEST_F()1239 auto* send_tensor = remote_enqueue_request.add_queue()->mutable_send_tensor(); in TEST_F() local1240 send_tensor->set_op_id(1); in TEST_F()1241 SetTensorProto(send_tensor->add_tensors()); in TEST_F()
421 auto* send_tensor = request.add_queue()->mutable_send_tensor(); in StartRemoteSendTensor() local422 send_tensor->set_op_id(recv_op_id_); in StartRemoteSendTensor()423 send_tensor->set_device_name(recv_device_->name()); in StartRemoteSendTensor()437 tensor.AsProtoTensorContent(send_tensor->add_tensors()); in StartRemoteSendTensor()
580 s = SendTensor(item.send_tensor(), context->Context()); in Enqueue()675 Status EagerServiceImpl::SendTensor(const SendTensorOp& send_tensor, in SendTensor() argument678 for (const auto& tensor_proto : send_tensor.tensors()) { in SendTensor()689 send_tensor.device_name().c_str(), &device)); in SendTensor()697 eager_context->RemoteMgr()->AddOperationOutputs(tensors, send_tensor.op_id()); in SendTensor()
214 Status SendTensor(const SendTensorOp& send_tensor,
64 SendTensorOp send_tensor = 3; field