| /external/tensorflow/tensorflow/core/kernels/data/sql/ |
| query_connection.h | 22 class IteratorContext;
61 virtual Status GetNext(IteratorContext* ctx, std::vector<Tensor>* out_tensors,
|
| sqlite_query_connection.h | 35 Status GetNext(IteratorContext* ctx, std::vector<Tensor>* out_tensors,
|
| sqlite_query_connection.cc | 52 Status SqliteQueryConnection::GetNext(IteratorContext* ctx,
60 // TODO(mrry): Pass in the `IteratorContext::allocator()`.
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| captured_function.h | 57 Status Run(IteratorContext* ctx, std::vector<Tensor>&& args,
63 Status RunWithBorrowedArgs(IteratorContext* ctx,
71 void RunAsync(IteratorContext* ctx, std::vector<Tensor>&& args,
93 Status MaybeInstantiate(IteratorContext* ctx,
|
| dataset_utils.h | 27 IteratorContext* ctx, const std::vector<Tensor>& input_element,
|
| dataset_utils.cc | 23 IteratorContext* ctx, const std::vector<Tensor>& input_element,
|
| skip_dataset_op.cc | 90 Status GetNextInternal(IteratorContext* ctx,
102 Status RestoreInternal(IteratorContext* ctx,
115 Status GetNextInternal(IteratorContext* ctx,
164 Status RestoreInternal(IteratorContext* ctx,
|
| take_dataset_op.cc | 91 Status GetNextInternal(IteratorContext* ctx,
103 Status RestoreInternal(IteratorContext* ctx,
116 Status GetNextInternal(IteratorContext* ctx,
151 Status RestoreInternal(IteratorContext* ctx,
|
| repeat_dataset_op.cc | 91 Status GetNextInternal(IteratorContext* ctx,
102 Status RestoreInternal(IteratorContext* ctx,
115 Status GetNextInternal(IteratorContext* ctx,
150 Status RestoreInternal(IteratorContext* ctx,
173 Status GetNextInternal(IteratorContext* ctx,
211 Status RestoreInternal(IteratorContext* ctx,
|
| prefetch_dataset_op.cc | 97 // through the IteratorContext to upstream,
106 Status GetNextInternal(IteratorContext* ctx,
175 Status RestoreInternal(IteratorContext* ctx,
222 Status EnsurePrefetchThreadStarted(IteratorContext* ctx)
228 new IteratorContext(*ctx))));
237 void PrefetchThread(IteratorContext* ctx) {
238 std::unique_ptr<IteratorContext> cleanup(ctx);
|
| tensor_dataset_op.cc | 92 Status GetNextInternal(IteratorContext* ctx,
115 Status RestoreInternal(IteratorContext* ctx,
|
| window_dataset.cc | 49 Status GetNextInternal(IteratorContext* ctx,
68 Status RestoreInternal(IteratorContext* ctx,
|
| stats_dataset_ops.cc | 92 Status GetNextInternal(IteratorContext* ctx,
114 Status RestoreInternal(IteratorContext* ctx,
188 Status GetNextInternal(IteratorContext* ctx,
212 Status RestoreInternal(IteratorContext* ctx,
|
| flat_map_dataset_op.cc | 131 Status GetNextInternal(IteratorContext* ctx,
197 Status RestoreInternal(IteratorContext* ctx,
239 Status BuildCurrentElementIteratorLocked(IteratorContext* ctx)
249 IteratorContext::Params params;
253 IteratorContext iter_ctx(std::move(params));
|
| map_and_batch_dataset_op.cc | 128 // through the IteratorContext to upstream,
156 Status GetNextInternal(IteratorContext* ctx,
247 void EnsureOutputAllocated(IteratorContext* ctx,
265 void InvokeFunctionLocked(IteratorContext* ctx, int64 batch_index,
285 IteratorContext* ctx, std::vector<Tensor> input_element) {
331 new IteratorContext(*ctx), std::move(input_element)));
334 void StartInvocationBatch(IteratorContext* ctx, int64 batch_index)
|
| random_dataset_op.cc | 98 Status GetNextInternal(IteratorContext* ctx,
117 Status RestoreInternal(IteratorContext* ctx,
|
| range_dataset_op.cc | 94 Status GetNextInternal(IteratorContext* ctx,
119 Status RestoreInternal(IteratorContext* ctx,
|
| tensor_slice_dataset_op.cc | 111 Status GetNextInternal(IteratorContext* ctx,
140 Status RestoreInternal(IteratorContext* ctx,
|
| zip_dataset_op.cc | 107 Status GetNextInternal(IteratorContext* ctx,
147 Status RestoreInternal(IteratorContext* ctx,
|
| captured_function.cc | 162 IteratorContext* ctx, FunctionLibraryRuntime::Handle* out_handle) {
191 Status CapturedFunction::Run(IteratorContext* ctx, std::vector<Tensor>&& args,
203 // TODO(mrry): Add cancellation manager support to IteratorContext
208 // will be required to plumb it through the `IteratorContext`.
224 Status CapturedFunction::RunWithBorrowedArgs(IteratorContext* ctx,
237 // TODO(mrry): Add cancellation manager support to IteratorContext
242 // will be required to plumb it through the `IteratorContext`.
259 void CapturedFunction::RunAsync(IteratorContext* ctx,
284 // TODO(mrry): Add cancellation manager support to IteratorContext
289 // will be required to plumb it through the `IteratorContext`
[all...] |
| cache_dataset_ops.cc | 115 Status GetNextInternal(IteratorContext* ctx,
217 Status GetNextInternal(IteratorContext* ctx,
311 Status GetNextInternal(IteratorContext* ctx,
346 Status GetNextInternal(IteratorContext* ctx,
374 Status GetNextInternal(IteratorContext* ctx,
|
| parallel_interleave_dataset_op.cc | 203 Status GetNextInternal(IteratorContext* ctx,
348 Status EnsureWorkerThreadsStarted(IteratorContext* ctx)
364 new IteratorContext(*ctx), i)));
378 void WorkerThread(IteratorContext* ctx_ptr, const int64 thread_index) {
381 std::unique_ptr<IteratorContext> ctx(ctx_ptr);
|
| batch_dataset_op.cc | 101 Status GetNextInternal(IteratorContext* ctx,
184 Status RestoreInternal(IteratorContext* ctx,
|
| /external/tensorflow/tensorflow/contrib/data/kernels/ |
| ignore_errors_dataset_op.cc | 78 Status GetNextInternal(IteratorContext* ctx,
111 Status RestoreInternal(IteratorContext* ctx,
|
| /external/tensorflow/tensorflow/core/framework/ |
| dataset.h | 246 // the IteratorContext?
253 class IteratorContext {
267 // about), and (ii) it can change after the `IteratorContext` has been
280 explicit IteratorContext(Params params) : params_(std::move(params)) {}
333 virtual Status GetNext(IteratorContext* ctx, std::vector<Tensor>* out_tensors,
352 virtual Status Restore(IteratorContext* ctx, IteratorStateReader* reader) {
368 Status RestoreParent(IteratorContext* ctx, IteratorStateReader* reader,
379 virtual Status RestoreInternal(IteratorContext* ctx,
514 Status GetNext(IteratorContext* ctx, std::vector<Tensor>* out_tensors,
536 virtual Status GetNextInternal(IteratorContext* ctx
[all...] |
