Searched refs:Conv2DWithBatchNorm (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 106 struct Conv2DWithBatchNorm { struct
107 Conv2DWithBatchNorm() = default;
108 Conv2DWithBatchNorm(const NodeDef* conv2d, const NodeDef* fused_batch_norm, in Conv2DWithBatchNorm() function
344 Conv2DWithBatchNorm* matched) { in FindConv2DWithBatchNorm()
403 Conv2DWithBatchNorm base; in FindConv2DWithBatchNormAndRelu()
578 const Conv2DWithBatchNorm& matched, GraphDef* optimized_graph, in AddFusedConv2DNode()
782 Conv2DWithBatchNorm conv2d_with_batch_norm; in Optimize()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_test.cc | 1173 static Conv2DWithBatchNormGraph Conv2DWithBatchNorm(int batch, int height, in Conv2DWithBatchNorm() function
1212 Conv2DWithBatchNormGraph conv_graph = Conv2DWithBatchNorm( in Conv2DWithBatchNormAndRelu()
1366 test::Benchmark(#type, Conv2DWithBatchNorm(N, H, W, C, FW, FH, FC).graph) \
|