Searched refs:allow_pruning_stateful_and_dataset_ops (Results 1 – 7 of 7) sorted by relevance
41 optimization_options.allow_pruning_stateful_and_dataset_ops = true; in CreateOptOptionsForEager()145 if (!optimization_options_.allow_pruning_stateful_and_dataset_ops) { in NodesToPreserve()153 if (!optimization_options_.allow_pruning_stateful_and_dataset_ops && in NodesToPreserve()
95 bool allow_pruning_stateful_and_dataset_ops = true; member
188 optimization_options.allow_pruning_stateful_and_dataset_ops = true; in Instantiate()
72 optimization_options().allow_pruning_stateful_and_dataset_ops = false; in GrapplerFunctionItem()
473 EXPECT_FALSE(opts.allow_pruning_stateful_and_dataset_ops); in TEST_F()
847 func_item.optimization_options().allow_pruning_stateful_and_dataset_ops = in OptimizeConsumeItem()
683 optimization_options.allow_pruning_stateful_and_dataset_ops = false; in Instantiate()