Searched refs:reduce_func_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | group_by_window_dataset_op.cc | 36 OP_REQUIRES_OK(ctx, ctx->GetAttr("reduce_func", &reduce_func_)); in GroupByWindowDatasetOp()
49 OP_REQUIRES_OK(ctx, CapturedFunction::Create(reduce_func_, ctx, in MakeDataset()
59 ctx, input, key_func_, reduce_func_, window_size_func_, in MakeDataset()
78 reduce_func_(reduce_func), in Dataset()
112 TF_RETURN_IF_ERROR(b->AddFunction(ctx, reduce_func_.name())); in AsGraphDefInternal()
139 b->BuildAttrValue(reduce_func_, &reduce_func); in AsGraphDefInternal()
517 const NameAttrList reduce_func_; member in tensorflow::data::__anon140756340111::GroupByWindowDatasetOp::Dataset
529 NameAttrList reduce_func_; member in tensorflow::data::__anon140756340111::GroupByWindowDatasetOp
|
| D | group_by_reducer_dataset_op.cc | 36 OP_REQUIRES_OK(ctx, ctx->GetAttr("reduce_func", &reduce_func_)); in GroupByReducerDatasetOp()
53 OP_REQUIRES_OK(ctx, CapturedFunction::Create(reduce_func_, ctx, in MakeDataset()
443 NameAttrList reduce_func_; member in tensorflow::data::__anon898653860111::GroupByReducerDatasetOp
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 643 OP_REQUIRES_OK(ctx, ctx->GetAttr("f", &reduce_func_)); in ReduceDatasetOp()
666 CapturedFunction::Create(reduce_func_, ctx, "other_arguments", in ComputeAsync()
749 NameAttrList reduce_func_; member in tensorflow::data::__anon9c98ff710411::ReduceDatasetOp
|