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.cc2780 class ConvertFusedBatchNorm : public OpRewritePattern<TF::FusedBatchNormOp> { class
2806 results.add<ConvertFusedBatchNorm>(context); in getCanonicalizationPatterns()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc4154 Status ConvertFusedBatchNorm(OpConverterParams* params) { in ConvertFusedBatchNorm() function
5815 REGISTER_DEFAULT_TRT_OP_CONVERTER(ConvertFusedBatchNorm,
Dconvert_nodes_test.cc2539 TEST_P(OpConverter_FP32_Test, ConvertFusedBatchNorm) { in TEST_P() argument