Home
last modified time | relevance | path

Searched refs:RestoreInput (Results 1 – 25 of 41) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/
Drepeat_dataset_op.cc188 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
267 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
Dflat_map_dataset_op.cc216 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
242 RestoreInput(ctx, reader, current_element_iterator_)); in RestoreInternal()
Dinterleave_dataset_op.cc244 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
299 RestoreInput(ctx, reader, current_elements_[idx])); in RestoreCurrentElements()
Dcache_dataset_ops.cc175 return RestoreInput(ctx, reader, iterator_); in RestoreInternal()
304 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
801 return RestoreInput(ctx, reader, iterator_); in RestoreInternal()
860 return RestoreInput(ctx, reader, input_impl_); in RestoreInternal()
Dfilter_by_component_dataset_op.cc159 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
Dtake_dataset_op.cc123 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
Dshard_dataset_op.cc169 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
Dzip_dataset_op.cc182 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl)); in RestoreInternal()
Dmodel_dataset_op.cc143 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
Dskip_dataset_op.cc189 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
Dconcatenate_dataset_op.cc179 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
Dbatch_dataset_op.cc236 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
Dgraph_rewrite_dataset.cc232 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dstats_dataset_ops.cc133 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
245 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
Ddirected_interleave_dataset_op.cc235 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, selector_input_impl_)); in RestoreInternal()
242 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, data_input_impls_[i])); in RestoreInternal()
Dchoose_fastest_dataset_op.cc263 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, fastest_input_impl_)); in RestoreInternal()
269 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl)); in RestoreInternal()
Dnon_serializable_dataset_op.cc109 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
Dsleep_dataset_op.cc123 return RestoreInput(ctx, reader, input_impl_); in RestoreInternal()
Dignore_errors_dataset_op.cc130 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
Dassert_next_dataset_op.cc140 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
Dchoose_fastest_branch_dataset_op.cc410 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
427 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, current_iterator_)); in RestoreInternal()
Dunique_dataset_op.cc144 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
Dgroup_by_window_dataset_op.cc350 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
398 RestoreInput(ctx, reader, current_group_iterator_)); in RestoreInternal()
Dunbatch_dataset_op.cc189 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()
Dtake_while_dataset_op.cc225 TF_RETURN_IF_ERROR(RestoreInput(ctx, reader, input_impl_)); in RestoreInternal()

12