Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/tensorflow/python/training/quantize_training.py: 67%

12 statements  

« prev     ^ index     » next       coverage.py v7.4.0, created at 2024-01-03 07:57 +0000

1# Copyright 2015 The TensorFlow Authors. All Rights Reserved. 

2# 

3# Licensed under the Apache License, Version 2.0 (the "License"); 

4# you may not use this file except in compliance with the License. 

5# You may obtain a copy of the License at 

6# 

7# http://www.apache.org/licenses/LICENSE-2.0 

8# 

9# Unless required by applicable law or agreed to in writing, software 

10# distributed under the License is distributed on an "AS IS" BASIS, 

11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 

12# See the License for the specific language governing permissions and 

13# limitations under the License. 

14# ============================================================================== 

15"""Quantize training for TensorFlow.""" 

16from tensorflow.core.framework import graph_pb2 

17from tensorflow.python._pywrap_quantize_training import DoQuantizeTrainingOnGraphDefHelper 

18from tensorflow.python.util import deprecation 

19from tensorflow.python.util.tf_export import tf_export 

20 

21 

22# Migrated this python code from deprecated quantize_training.i 

23@deprecation.deprecated( 

24 None, 

25 "GraphDef quantized training rewriter is deprecated in the long term.") 

26@tf_export(v1=["train.do_quantize_training_on_graphdef"]) 

27def do_quantize_training_on_graphdef(input_graph, num_bits): 

28 """A general quantization scheme is being developed in `tf.contrib.quantize`. 

29 

30 Consider using that instead, though since it is in the tf.contrib namespace, 

31 it is not subject to backward compatibility guarantees. 

32 

33 Args: 

34 input_graph: A `GraphDef`. 

35 num_bits: The number of bits for quantize training. 

36 

37 Returns: 

38 The graph with quantize training done. 

39 """ 

40 

41 graph = graph_pb2.GraphDef() 

42 result_graph_string = DoQuantizeTrainingOnGraphDefHelper( 

43 input_graph.SerializeToString(), num_bits) 

44 

45 graph.ParseFromString(result_graph_string) 

46 return graph