Searched refs:merged_graph_def (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes.cc | 220 GraphDef merged_graph_def; in MergeDuplicateNodes() local
236 NodeDef* new_node = merged_graph_def.mutable_node()->Add(); in MergeDuplicateNodes()
243 TF_RETURN_IF_ERROR(RenameNodeInputs(merged_graph_def, inputs_to_rename, in MergeDuplicateNodes()
924 GraphDef merged_graph_def; in QuantizeNodes() local
926 &merged_graph_def)); in QuantizeNodes()
927 TF_RETURN_IF_ERROR(IsGraphValid(merged_graph_def)); in QuantizeNodes()
933 MergeDuplicateNodes(merged_graph_def, context, &deduped_graph_def)); in QuantizeNodes()
|
| D | quantize_nodes_test.cc | 1218 GraphDef merged_graph_def; in TestMergeDuplicateQuantizes() local
1220 {}, {"add_op"}, {}, 1.0, &merged_graph_def); in TestMergeDuplicateQuantizes()
1223 for (const NodeDef& node : merged_graph_def.node()) { in TestMergeDuplicateQuantizes()
1254 GraphDef merged_graph_def; in TestMergeDuplicateConsts() local
1256 {}, {"mul_op"}, {}, 1.0, &merged_graph_def); in TestMergeDuplicateConsts()
1259 MapNamesToNodes(merged_graph_def, &node_map); in TestMergeDuplicateConsts()
1303 GraphDef merged_graph_def; in TestMergeDuplicatesNested() local
1305 {}, {"mul_op"}, {}, 1.0, &merged_graph_def); in TestMergeDuplicatesNested()
1308 MapNamesToNodes(merged_graph_def, &node_map); in TestMergeDuplicatesNested()
1358 GraphDef merged_graph_def; in TestMergeDuplicatesInOut() local
[all …]
|