Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes.cc553 std::vector<NodeDef> linear_nodes; in HoistFakeQuants() local
556 linear_nodes.push_back(current_match.inputs[0].node); in HoistFakeQuants()
563 0, linear_nodes[linear_nodes.size() - 2].input(0)); in HoistFakeQuants()
569 linear_nodes[linear_nodes.size() - 2].set_input( in HoistFakeQuants()
571 linear_nodes.front().set_name(fake_quant_node.name()); in HoistFakeQuants()
572 for (const NodeDef& linear_node : linear_nodes) { in HoistFakeQuants()