Searched refs:reduce_dataset_op (Results 1 – 3 of 3) sorted by relevance
63 "data/reduce_dataset_op.cc",
1012 name = "reduce_dataset_op",1013 srcs = ["reduce_dataset_op.cc"],1014 hdrs = ["reduce_dataset_op.h"],1033 ":reduce_dataset_op",1584 ":reduce_dataset_op",
2044 if (auto reduce_dataset_op = dyn_cast<ReduceDatasetOp>(op)) { in InferShapeForSingleOperation() local2046 return InferShapeForReduceDataset(reduce_dataset_op, max_iterations); in InferShapeForSingleOperation()