Searched refs:queue_runner_def (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | queue_runner_impl.py | 63 queue_runner_def=None, import_scope=None): argument
102 if queue_runner_def:
105 self._init_from_proto(queue_runner_def,
168 def _init_from_proto(self, queue_runner_def, import_scope=None): argument
175 assert isinstance(queue_runner_def, queue_runner_pb2.QueueRunnerDef)
178 ops.prepend_name_scope(queue_runner_def.queue_name, import_scope))
181 for op in queue_runner_def.enqueue_op_name]
183 queue_runner_def.close_op_name, import_scope))
185 queue_runner_def.cancel_op_name, import_scope))
188 for code in queue_runner_def.queue_closed_exception_types)
[all …]
|
| /external/tensorflow/tensorflow/cc/training/ |
| D | queue_runner_test.cc | 87 QueueRunnerDef queue_runner_def; in BuildQueueRunnerDef() local
88 *queue_runner_def.mutable_queue_name() = queue_name; in BuildQueueRunnerDef()
90 *queue_runner_def.mutable_enqueue_op_name()->Add() = enqueue_op; in BuildQueueRunnerDef()
92 *queue_runner_def.mutable_close_op_name() = close_op; in BuildQueueRunnerDef()
93 *queue_runner_def.mutable_cancel_op_name() = cancel_op; in BuildQueueRunnerDef()
95 *queue_runner_def.mutable_queue_closed_exception_types()->Add() = in BuildQueueRunnerDef()
98 return queue_runner_def; in BuildQueueRunnerDef()
115 QueueRunnerDef queue_runner_def = BuildQueueRunnerDef( in TEST() local
119 TF_EXPECT_OK(QueueRunner::New(queue_runner_def, &qr)); in TEST()
134 QueueRunnerDef queue_runner_def = BuildQueueRunnerDef( in TEST() local
[all …]
|
| D | queue_runner.cc | 22 Status QueueRunner::New(const QueueRunnerDef& queue_runner_def, in New() argument
25 return (*result)->Init(queue_runner_def); in New()
28 Status QueueRunner::New(const QueueRunnerDef& queue_runner_def, in New() argument
33 return (*result)->Init(queue_runner_def); in New()
46 Status QueueRunner::Init(const QueueRunnerDef& queue_runner_def) { in Init() argument
47 queue_name_ = queue_runner_def.queue_name(); in Init()
50 queue_runner_def.enqueue_op_name().begin(), in Init()
51 queue_runner_def.enqueue_op_name().end()); in Init()
61 close_op_name_ = queue_runner_def.close_op_name(); in Init()
62 cancel_op_name_ = queue_runner_def.cancel_op_name(); in Init()
[all …]
|
| D | queue_runner.h | 43 static Status New(const QueueRunnerDef& queue_runner_def,
47 static Status New(const QueueRunnerDef& queue_runner_def, Coordinator* coord,
92 Status Init(const QueueRunnerDef& queue_runner_def);
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.-queue-runner.pbtxt | 35 …', \'close_op\', \'cancel_op\', \'queue_closed_exception_types\', \'queue_runner_def\', \'import_s…
43 …argspec: "args=[\'queue_runner_def\', \'import_scope\'], varargs=None, keywords=None, defaults=[\'…
|
| D | tensorflow.train.queue_runner.-queue-runner.pbtxt | 35 …', \'close_op\', \'cancel_op\', \'queue_closed_exception_types\', \'queue_runner_def\', \'import_s…
43 …argspec: "args=[\'queue_runner_def\', \'import_scope\'], varargs=None, keywords=None, defaults=[\'…
|