Searched refs:init_func_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | generator_dataset_op.cc | 39 init_func_(std::move(init_func)), in Dataset()
88 dataset()->init_func_->Instantiate(ctx, &instantiated_init_func_)); in Initialize()
148 const std::unique_ptr<CapturedFunction> init_func_; member in tensorflow::data::GeneratorDatasetOp::Dataset
157 OP_REQUIRES_OK(ctx, ctx->GetAttr("init_func", &init_func_)); in GeneratorDatasetOp()
168 init_func_, ctx, "init_func_other_args", &init_func)); in MakeDataset()
|
| D | generator_dataset_op.h | 35 NameAttrList init_func_; variable
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | group_by_reducer_dataset_op.cc | 35 OP_REQUIRES_OK(ctx, ctx->GetAttr("init_func", &init_func_)); in GroupByReducerDatasetOp()
49 OP_REQUIRES_OK(ctx, CapturedFunction::Create(init_func_, ctx, in MakeDataset()
442 NameAttrList init_func_; member in tensorflow::data::__anon898653860111::GroupByReducerDatasetOp
|