Home
last modified time | relevance | path

Searched refs:ConvertAvgPoolOp (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc767 class ConvertAvgPoolOp : public OpConversionPattern<mhlo::DivOp> { class
1254 ->insert<ConvertAvgPoolOp, ConvertConvOp, ConvertDynamicSliceOp, in PopulateLegalizeHloToTfPatterns()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc2256 class ConvertAvgPoolOp : public OpRewritePattern<OpTy> { class
2319 using ConvertAvgPool2DOp = ConvertAvgPoolOp<TF::AvgPoolOp, /*num_dims=*/4>;
2320 using ConvertAvgPool3DOp = ConvertAvgPoolOp<TF::AvgPool3DOp, /*num_dims=*/5>;