Home
last modified time | relevance | path

Searched refs:NodesToPreserve (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item.h82 std::unordered_set<string> NodesToPreserve() const;
Dgrappler_item.cc80 std::unordered_set<string> GrapplerItem::NodesToPreserve() const { in NodesToPreserve() function in tensorflow::grappler::GrapplerItem
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dtrt_optimization_pass.cc240 for (const auto& n : item.NodesToPreserve()) { in Optimize()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer.cc719 TF_RETURN_IF_ERROR(RemoveStackOps(item.NodesToPreserve(), optimized_graph)); in Optimize()
729 TF_RETURN_IF_ERROR(RemoveDeadBranches(item.NodesToPreserve(), node_map, in Optimize()
Dmodel_pruner.cc425 const std::unordered_set<string> nodes_to_preserve = item.NodesToPreserve(); in Optimize()
Ddependency_optimizer.cc679 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
Dremapper.cc51 : nodes_to_preserve(item.NodesToPreserve()), in RemapperContext()
Dscoped_allocator_optimizer.cc681 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
Dlayout_optimizer.cc2211 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
Dconstant_folding.cc3428 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
Darithmetic_optimizer.cc3653 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()