Searched refs:keep_ops (Results 1 – 9 of 9) sorted by relevance
39 item.keep_ops = keep_ops; in WithGraph()93 for (const auto& node : keep_ops) { in NodesToPreserve()
70 std::vector<string> keep_ops; member
435 new_item->keep_ops.push_back(NodeName(node)); in GrapplerItemFromMetaGraphDef()
165 if (!item.keep_ops.empty()) { in PrintDebugInfo()167 for (const auto& f : item.keep_ops) { in PrintDebugInfo()
672 ASSERT_EQ(1, item.keep_ops.size()); in TEST_F()673 EXPECT_EQ("update", item.keep_ops[0]); in TEST_F()
340 keep_ops.push_back(control_output.node_name); in GrapplerFunctionItem()
572 item.keep_ops.push_back("stack1"); in TEST_F()573 item.keep_ops.push_back("stack2"); in TEST_F()
739 item.keep_ops = std::move(keep_node_names); in OptimizeGraph()
332 return absl::c_any_of(item_->keep_ops, [&](const string& keep_node) { in IsKeepOp()