Searched refs:finalize_func_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | generator_dataset_op.cc | 41 finalize_func_(std::move(finalize_func)), in Dataset()
91 TF_RETURN_IF_ERROR(dataset()->finalize_func_->Instantiate( in Initialize()
150 const std::unique_ptr<CapturedFunction> finalize_func_; member in tensorflow::data::GeneratorDatasetOp::Dataset
159 OP_REQUIRES_OK(ctx, ctx->GetAttr("finalize_func", &finalize_func_)); in GeneratorDatasetOp()
175 OP_REQUIRES_OK(ctx, CapturedFunction::Create(finalize_func_, ctx, in MakeDataset()
|
| D | generator_dataset_op.h | 37 NameAttrList finalize_func_; variable
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | group_by_reducer_dataset_op.cc | 37 OP_REQUIRES_OK(ctx, ctx->GetAttr("finalize_func", &finalize_func_)); in GroupByReducerDatasetOp()
58 CapturedFunction::Create(finalize_func_, ctx, in MakeDataset()
444 NameAttrList finalize_func_; member in tensorflow::data::__anon898653860111::GroupByReducerDatasetOp
|