Home
last modified time | relevance | path

Searched refs:FoldBatchNorms (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_batch_norms_test.cc33 Status FoldBatchNorms(const GraphDef& input_graph_def,
76 FoldBatchNorms(original_graph_def, {{}, {"output"}}, &fused_graph_def)); in TestFoldBatchNormsConv2D()
127 FoldBatchNorms(original_graph_def, {{}, {"output"}}, &fused_graph_def)); in TestFoldBatchNormsDepthwiseConv2dNative()
186 TF_ASSERT_OK(FoldBatchNorms( in TestFoldBatchNormsConv2DShared()
235 FoldBatchNorms(original_graph_def, {{}, {"output"}}, &fused_graph_def)); in TestFoldBatchNormsMatMul()
Dfold_batch_norms.cc32 Status FoldBatchNorms(const GraphDef& input_graph_def, in FoldBatchNorms() function
123 REGISTER_GRAPH_TRANSFORM("fold_batch_norms", FoldBatchNorms);
/external/tensorflow/tensorflow/contrib/quantize/python/
Dfold_batch_norms_test.py138 fold_batch_norms.FoldBatchNorms(
231 fold_batch_norms.FoldBatchNorms(
309 fold_batch_norms.FoldBatchNorms(
403 fold_batch_norms.FoldBatchNorms(
501 fold_batch_norms.FoldBatchNorms(
613 fold_batch_norms.FoldBatchNorms(
713 fold_batch_norms.FoldBatchNorms(
Dquantize_parameterized_test.py536 fold_batch_norms.FoldBatchNorms(graph, is_training=True)
590 fold_batch_norms.FoldBatchNorms(graph, is_training=True)
649 fold_batch_norms.FoldBatchNorms(graph, is_training=True)
708 fold_batch_norms.FoldBatchNorms(graph, is_training=True)
720 fold_batch_norms.FoldBatchNorms(graph, is_training=True)
775 fold_batch_norms.FoldBatchNorms(graph, is_training=True)
Dquantize_graph.py70 fold_batch_norms.FoldBatchNorms(
Dfold_batch_norms.py37 def FoldBatchNorms(graph, is_training, freeze_batch_norm_delay=None): function