Searched refs:quantized_graph_def (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_weights_test.cc | 76 GraphDef quantized_graph_def; in TestQuantizeWeights() local
78 QuantizeWeights(original_graph_def, context, &quantized_graph_def)); in TestQuantizeWeights()
82 MapNamesToNodes(quantized_graph_def, &node_lookup); in TestQuantizeWeights()
104 TF_ASSERT_OK(quantized_session->Create(quantized_graph_def)); in TestQuantizeWeights()
131 GraphDef quantized_graph_def; in TEST_F() local
133 QuantizeWeights(original_graph_def, context, &quantized_graph_def)); in TEST_F()
136 MapNamesToNodes(quantized_graph_def, &node_lookup); in TEST_F()
|
| D | quantize_nodes_test.cc | 102 GraphDef quantized_graph_def; in TestQuantizedVersusFloatGraph() local
105 &quantized_graph_def); in TestQuantizedVersusFloatGraph()
111 for (const NodeDef& node : quantized_graph_def.node()) { in TestQuantizedVersusFloatGraph()
136 GraphDef quantized_graph_def; in TestGraphWithInputRange() local
139 output_names, context, 1.0, &quantized_graph_def); in TestGraphWithInputRange()
146 GraphDef* quantized_graph_def) { in TestGraphWithFallbackRange() argument
152 context, 2.0, quantized_graph_def); in TestGraphWithFallbackRange()
186 GraphDef quantized_graph_def; in TestIgnoreOps() local
189 &quantized_graph_def); in TestIgnoreOps()
195 for (const NodeDef& node : quantized_graph_def.node()) { in TestIgnoreOps()
[all …]
|
| D | quantize_nodes.cc | 692 GraphDef quantized_graph_def; in QuantizeNodes() local
918 {}, &quantized_graph_def)); in QuantizeNodes()
919 TF_RETURN_IF_ERROR(IsGraphValid(quantized_graph_def)); in QuantizeNodes()
925 TF_RETURN_IF_ERROR(MergeAdjacentRequantizes(quantized_graph_def, context, in QuantizeNodes()
|