Home
last modified time | relevance | path

Searched refs:ops_to_ignore (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes.cc636 std::set<string> ops_to_ignore; in QuantizeNodes() local
639 ops_to_ignore.insert(name); in QuantizeNodes()
648 if (ops_to_ignore.count(op_info.float_name) == 0) { in QuantizeNodes()
Dquantize_nodes_test.cc155 void TestIgnoreOps(std::initializer_list<string> ops_to_ignore) { in TestIgnoreOps() argument
182 if (ops_to_ignore.size() > 0) { in TestIgnoreOps()
183 context.params["ignore_op"] = ops_to_ignore; in TestIgnoreOps()
193 for (const string& op_name : ops_to_ignore) { in TestIgnoreOps()