Searched refs:func_item (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 598 GrapplerFunctionItem func_item; in Optimize() local
600 func, flib, trimmed_item.graph.versions().producer(), &func_item)); in Optimize()
606 func_item.optimization_options().allow_non_differentiable_rewrites = in Optimize()
611 Status added_devices = func_item.AddDevices(item); in Optimize()
620 func_item.optimization_options().allow_pruning_stateful_and_dataset_ops = in Optimize()
626 OptimizeGraph(cluster, func_item, &optimized_func_graph)); in Optimize()
639 func_item.SwapFunctionBody(std::move(optimized_func_graph)); in Optimize()
640 TF_RETURN_IF_ERROR(MakeFunctionDef(func_item, flib, &optimized_func)); in Optimize()
|
| D | function_optimizer.cc | 1327 GrapplerFunctionItem func_item; in MaybeDeadOutputs() local
1330 ctx.function_library(), ctx.graph_version(), &func_item)); in MaybeDeadOutputs()
1333 TF_RETURN_IF_ERROR(MaybeDeadOutputs(ctx, func_item, &func_dead_outputs)); in MaybeDeadOutputs()
|