| 81 | } |
| 82 | |
| 83 | QueueRunnerDef BuildQueueRunnerDef( |
| 84 | const std::string& queue_name, const std::vector<std::string>& enqueue_ops, |
| 85 | const std::string& close_op, const std::string& cancel_op, |
| 86 | const std::vector<Code>& queue_closed_error_codes) { |
| 87 | QueueRunnerDef queue_runner_def; |
| 88 | *queue_runner_def.mutable_queue_name() = queue_name; |
| 89 | for (const std::string& enqueue_op : enqueue_ops) { |
| 90 | *queue_runner_def.mutable_enqueue_op_name()->Add() = enqueue_op; |
| 91 | } |
| 92 | *queue_runner_def.mutable_close_op_name() = close_op; |
| 93 | *queue_runner_def.mutable_cancel_op_name() = cancel_op; |
| 94 | for (const auto& error_code : queue_closed_error_codes) { |
| 95 | *queue_runner_def.mutable_queue_closed_exception_types()->Add() = |
| 96 | error_code; |
| 97 | } |
| 98 | return queue_runner_def; |
| 99 | } |
| 100 | |
| 101 | std::unique_ptr<Session> BuildSessionAndInitVariable( |
| 102 | const GraphDef& graph_def) { |