Searched refs:background_worker_ (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | to_tf_record_op.cc | 35 background_worker_(ctx->env(), "tf_data_to_tf_record") {} in ToTFRecordOp() 52 background_worker_.Schedule([this, ctx, done = std::move(done)]() { in ComputeAsync() 106 BackgroundWorker background_worker_; member in tensorflow::data::experimental::__anone30cd1e90111::ToTFRecordOp
|
/external/tensorflow/tensorflow/core/kernels/ |
D | lookup_table_init_op.cc | 173 background_worker_(ctx->env(), "initialize_table_from_dataset") {} in InitializeTableFromDatasetOp() 183 background_worker_.Schedule([ctx, dataset, table, done]() { in ComputeAsync() 191 data::BackgroundWorker background_worker_; member in tensorflow::InitializeTableFromDatasetOp
|
/external/tensorflow/tensorflow/core/kernels/data/ |
D | iterator_ops.cc | 582 background_worker_(ctx->env(), background_worker_name) {} in HybridAsyncOpKernel() 586 background_worker_.Schedule([this, ctx, done = std::move(done)]() { in ComputeAsync() 774 background_worker_(ctx->env(), "tf_data_one_shot_iterator"), in OneShotIteratorOp() 818 background_worker_.Schedule([this, ctx, done]() { Init(ctx, done); }); in ComputeAsync() 937 BackgroundWorker background_worker_; member in tensorflow::data::__anonadcc0d0f0711::OneShotIteratorOp
|
D | multi_device_iterator_ops.cc | 598 background_worker_(ctx->env(), in MultiDeviceIteratorGetNextFromShardOp() 615 background_worker_.Schedule(std::bind( in ComputeAsync() 650 BackgroundWorker background_worker_; member in tensorflow::data::__anon2e5a99280111::MultiDeviceIteratorGetNextFromShardOp
|
D | iterator_ops.h | 222 BackgroundWorker background_worker_;
|