Searched refs:HybridAsyncOpKernel (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.h | 210 class HybridAsyncOpKernel : public AsyncOpKernel {
212 HybridAsyncOpKernel(OpKernelConstruction* ctx,
225 class MakeIteratorOp : public HybridAsyncOpKernel {
228 : HybridAsyncOpKernel(ctx, "tf_data_make_iterator") {} in MakeIteratorOp()
234 class IteratorGetNextOp : public HybridAsyncOpKernel {
237 : HybridAsyncOpKernel(ctx, "tf_data_iterator_get_next") { in IteratorGetNextOp()
252 class DeleteIteratorOp : public HybridAsyncOpKernel {
255 : HybridAsyncOpKernel(ctx, "tf_data_delete_iterator") {} in DeleteIteratorOp()
261 class IteratorGetNextAsOptionalOp : public HybridAsyncOpKernel {
264 : HybridAsyncOpKernel(ctx, "tf_data_iterator_get_next_as_optional") { in IteratorGetNextAsOptionalOp()
|
| D | iterator_ops.cc | 579 HybridAsyncOpKernel::HybridAsyncOpKernel(OpKernelConstruction* ctx, in HybridAsyncOpKernel() function in tensorflow::data::HybridAsyncOpKernel
584 void HybridAsyncOpKernel::ComputeAsync(OpKernelContext* ctx, in ComputeAsync()
592 void HybridAsyncOpKernel::Compute(OpKernelContext* ctx) { in Compute()
620 class ToSingleElementOp : public HybridAsyncOpKernel {
623 : HybridAsyncOpKernel(ctx, "tf_data_to_single_element") { in ToSingleElementOp()
678 class ReduceDatasetOp : public HybridAsyncOpKernel {
681 : HybridAsyncOpKernel(ctx, "tf_data_reduce_dataset") { in ReduceDatasetOp()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | io_ops.h | 30 class SaveDatasetOp : public HybridAsyncOpKernel {
|
| D | io_ops.cc | 37 : HybridAsyncOpKernel(ctx, "tf_data_save_dataset") { in SaveDatasetOp()
|