Searched refs:GenericConvertReductionOp (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 4105 class GenericConvertReductionOp : public OpRewritePattern<OpTy> { class
4219 : public GenericConvertReductionOp<ConvertMeanOp, TF::MeanOp, AddOp> {
4221 using GenericConvertReductionOp::GenericConvertReductionOp;
4234 : public GenericConvertReductionOp<ConvertSumOp, TF::SumOp, AddOp> {
4236 using GenericConvertReductionOp::GenericConvertReductionOp;
4251 : public GenericConvertReductionOp<ConvertMaxOp, TF::MaxOp, MaxOp,
4254 using GenericConvertReductionOp::GenericConvertReductionOp;
4269 : public GenericConvertReductionOp<ConvertMinOp, TF::MinOp, MinOp,
4272 using GenericConvertReductionOp::GenericConvertReductionOp;
4287 : public GenericConvertReductionOp<ConvertProdOp, TF::ProdOp, MulOp> {
[all …]
|