Searched refs:FoldBatchNorms (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_batch_norms_test.cc | 33 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()
|
| D | fold_batch_norms.cc | 31 Status FoldBatchNorms(const GraphDef& input_graph_def, in FoldBatchNorms() function
122 REGISTER_GRAPH_TRANSFORM("fold_batch_norms", FoldBatchNorms);
|