Searched refs:NodesToPreserve (Results 1 – 15 of 15) sorted by relevance
83 std::unordered_set<string> NodesToPreserve() const;
106 std::unordered_set<string> GrapplerItem::NodesToPreserve() const { in NodesToPreserve() function in tensorflow::grappler::GrapplerItem
194 const auto nodes_to_preserve = item.NodesToPreserve(); in OptimizeAndCollectStats()
274 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
746 TF_RETURN_IF_ERROR(RemoveStackOps(item.NodesToPreserve(), optimized_graph)); in Optimize()754 TF_RETURN_IF_ERROR(RemoveDeadBranches(item.NodesToPreserve(), node_map, in Optimize()
423 const std::unordered_set<string> nodes_to_preserve = item.NodesToPreserve(); in Optimize()
745 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
917 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
80 : nodes_to_preserve(item->NodesToPreserve()), in RemapperContext()
2014 AutoMixedPrecisionImpl optimizer(cluster, item.NodesToPreserve(), output, in Optimize()
2272 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
238 const auto& nodes_to_preserve = item.NodesToPreserve(); in InitializeTransposeContext()
3791 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
3981 nodes_to_preserve_ = item.NodesToPreserve(); in Optimize()
215 for (const auto& n : item.NodesToPreserve()) { in Optimize()