Searched refs:keep_ops (Results 1 – 9 of 9) sorted by relevance
58 item.keep_ops = keep_ops; in WithGraph()119 for (const auto& node : keep_ops) { in NodesToPreserve()
71 std::vector<string> keep_ops; member
542 new_item->keep_ops.push_back(NodeName(node)); in GrapplerItemFromMetaGraphDef()
175 if (!item.keep_ops.empty()) { in PrintDebugInfo()177 for (const auto& f : item.keep_ops) { in PrintDebugInfo()
499 ASSERT_EQ(1, item.keep_ops.size()); in TEST_F()500 EXPECT_EQ("update", item.keep_ops[0]); in TEST_F()
67 keep_ops.push_back(control_output.node_name); in GrapplerFunctionItem()
1222 NodeNames keep_nodes(item.keep_ops.begin(), item.keep_ops.end()); in InlineFunctionCalls()
687 item.keep_ops.push_back("stack1"); in TEST_F()688 item.keep_ops.push_back("stack2"); in TEST_F()
1000 item.keep_ops = std::move(keep_node_names); in OptimizeGraph()