Searched refs:freeze_batch_norm_delay (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| D | fold_batch_norms_test.py | 75 fused_batch_norm, freeze_batch_norm_delay, argument
139 g, is_training=True, freeze_batch_norm_delay=freeze_batch_norm_delay)
163 if freeze_batch_norm_delay is not None:
177 freeze_batch_norm_delay=None, argument
232 g, is_training=True, freeze_batch_norm_delay=freeze_batch_norm_delay)
256 if freeze_batch_norm_delay is not None:
268 freeze_batch_norm_delay, argument
310 g, is_training=True, freeze_batch_norm_delay=freeze_batch_norm_delay)
333 if freeze_batch_norm_delay is not None:
344 freeze_batch_norm_delay, insert_identity_node): argument
[all …]
|
| D | fold_batch_norms.py | 37 def FoldBatchNorms(graph, is_training, freeze_batch_norm_delay=None): argument
53 graph, is_training, freeze_batch_norm_delay=freeze_batch_norm_delay)
57 freeze_batch_norm_delay=freeze_batch_norm_delay)
60 def _FoldFusedBatchNorms(graph, is_training, freeze_batch_norm_delay): argument
98 freeze_batch_norm_delay=freeze_batch_norm_delay))
344 def _ComputeBatchNormCorrections(context, match, freeze_batch_norm_delay): argument
398 if freeze_batch_norm_delay is not None:
401 freeze_batch_norm_delay,
502 def _FoldUnfusedBatchNorms(graph, is_training, freeze_batch_norm_delay): argument
530 freeze_batch_norm_delay=freeze_batch_norm_delay,
[all …]
|
| D | quantize_graph.py | 72 freeze_batch_norm_delay=freeze_bn_delay,
|