Home
last modified time | relevance | path

Searched refs:ConvertFusedBatchNorm (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc2587 class ConvertFusedBatchNorm : public OpRewritePattern<TF::FusedBatchNormOp> { class
2613 results.insert<ConvertFusedBatchNorm>(context); in getCanonicalizationPatterns()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc5161 Status ConvertFusedBatchNorm(OpConverterParams* params) { in ConvertFusedBatchNorm() function
6290 (*registration)[normalization_op_type] = ConvertFusedBatchNorm; in RegisterValidatableOpConverters()
Dconvert_nodes_test.cc2149 TEST_P(OpConverter_FP32_Test, ConvertFusedBatchNorm) { in TEST_P() argument