Searched refs:quantize_graph (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/tools/quantization/ |
H A D | quantize_graph_test.py | 35 from tensorflow.tools.quantization import quantize_graph namespace 57 a_constant = quantize_graph.create_constant_node( 60 b_constant = quantize_graph.create_constant_node( 63 mat_mul_node = quantize_graph.create_node("MatMul", mat_mul_name, 65 quantize_graph.set_attr_dtype(mat_mul_node, "T", dtypes.float32) 66 quantize_graph.set_attr_bool(mat_mul_node, "transpose_a", False) 67 quantize_graph.set_attr_bool(mat_mul_node, "transpose_b", False) 81 input_constant = quantize_graph.create_constant_node( 87 filter_constant = quantize_graph.create_constant_node( 93 conv_node = quantize_graph [all...] |
/external/tensorflow/tensorflow/contrib/quantize/ |
H A D | __init__.py | 22 from tensorflow.contrib.quantize.python.quantize_graph import *
|
/external/tensorflow/tensorflow/contrib/quantize/python/ |
H A D | quantize_graph_test.py | 15 """Unit tests for the quantize_graph graph rewriting API.""" 22 from tensorflow.contrib.quantize.python import quantize_graph namespace 33 # just the specific features of the quantize_graph API. 37 quantize_graph.create_training_graph, 38 quantize_graph.create_eval_graph, 39 quantize_graph.experimental_create_training_graph, 40 quantize_graph.experimental_create_eval_graph, 47 quantize_graph.create_training_graph, 48 quantize_graph.experimental_create_training_graph, 55 quantize_graph [all...] |
Completed in 2609 milliseconds