Home
last modified time | relevance | path

Searched refs:allow_pruning_stateful_and_dataset_ops (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/
H A Dgrappler_item.cc41 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()
H A Dgrappler_item.h95 bool allow_pruning_stateful_and_dataset_ops = true; member
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dpartitioned_function_ops.cc189 optimization_options.allow_pruning_stateful_and_dataset_ops = true; in Instantiate()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dfunctions.cc72 optimization_options().allow_pruning_stateful_and_dataset_ops = false; in GrapplerFunctionItem()
H A Dfunctions_test.cc473 EXPECT_FALSE(opts.allow_pruning_stateful_and_dataset_ops); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dcaptured_function.cc639 optimization_options.allow_pruning_stateful_and_dataset_ops = false; in Instantiate()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmeta_optimizer.cc1214 func_item.optimization_options().allow_pruning_stateful_and_dataset_ops = in OptimizeConsumeItem()