Searched refs:send_tensor (Results 1 – 5 of 5) sorted by relevance
1063 auto* send_tensor = remote_enqueue_request.add_queue()->mutable_send_tensor(); in TEST_F() local1064 send_tensor->set_op_id(1); in TEST_F()1065 SetTensorProto(send_tensor->add_tensors()); in TEST_F()1136 auto* send_tensor = remote_enqueue_request.add_queue()->mutable_send_tensor(); in TEST_F() local1137 send_tensor->set_op_id(1); in TEST_F()1138 SetTensorProto(send_tensor->add_tensors()); in TEST_F()1146 remote_handle->set_op_id(send_tensor->op_id()); in TEST_F()1241 auto* send_tensor = remote_enqueue_request.add_queue()->mutable_send_tensor(); in TEST_F() local1242 send_tensor->set_op_id(1); in TEST_F()1243 SetTensorProto(send_tensor->add_tensors()); in TEST_F()
427 auto* send_tensor = request.add_queue()->mutable_send_tensor(); in StartRemoteSendTensor() local428 send_tensor->set_op_id(recv_op_id_); in StartRemoteSendTensor()429 send_tensor->set_device_name(recv_device_->name()); in StartRemoteSendTensor()441 tensor.AsProtoTensorContent(send_tensor->add_tensors()); in StartRemoteSendTensor()
604 s = SendTensor(item.send_tensor(), context->Context()); in Enqueue()698 Status EagerServiceImpl::SendTensor(const SendTensorOp& send_tensor, in SendTensor() argument701 for (const auto& tensor_proto : send_tensor.tensors()) { in SendTensor()712 send_tensor.device_name().c_str(), &device)); in SendTensor()720 eager_context->RemoteMgr()->AddOperationOutputs(tensors, send_tensor.op_id()); in SendTensor()
215 Status SendTensor(const SendTensorOp& send_tensor,
64 SendTensorOp send_tensor = 3; field