Home
last modified time | relevance | path

Searched refs:keep_ops (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item.cc58 item.keep_ops = keep_ops; in WithGraph()
119 for (const auto& node : keep_ops) { in NodesToPreserve()
Dgrappler_item.h71 std::vector<string> keep_ops; member
Dgrappler_item_builder.cc542 new_item->keep_ops.push_back(NodeName(node)); in GrapplerItemFromMetaGraphDef()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dtrt_optimization_pass.cc175 if (!item.keep_ops.empty()) { in PrintDebugInfo()
177 for (const auto& f : item.keep_ops) { in PrintDebugInfo()
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions_test.cc499 ASSERT_EQ(1, item.keep_ops.size()); in TEST_F()
500 EXPECT_EQ("update", item.keep_ops[0]); in TEST_F()
Dfunctions.cc67 keep_ops.push_back(control_output.node_name); in GrapplerFunctionItem()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc1222 NodeNames keep_nodes(item.keep_ops.begin(), item.keep_ops.end()); in InlineFunctionCalls()
Dloop_optimizer_test.cc687 item.keep_ops.push_back("stack1"); in TEST_F()
688 item.keep_ops.push_back("stack2"); in TEST_F()
Dmeta_optimizer.cc1000 item.keep_ops = std::move(keep_node_names); in OptimizeGraph()