Searched refs:GenericConvertReductionOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 3616 class GenericConvertReductionOp : public OpRewritePattern<OpTy> { class
3711 : public GenericConvertReductionOp<ConvertMeanOp, TF::MeanOp, AddOp> {
3713 using GenericConvertReductionOp::GenericConvertReductionOp;
3726 : public GenericConvertReductionOp<ConvertSumOp, TF::SumOp, AddOp> {
3728 using GenericConvertReductionOp::GenericConvertReductionOp;
3742 : public GenericConvertReductionOp<ConvertMaxOp, TF::MaxOp, MaxOp,
3745 using GenericConvertReductionOp::GenericConvertReductionOp;
3760 : public GenericConvertReductionOp<ConvertMinOp, TF::MinOp, MinOp,
3763 using GenericConvertReductionOp::GenericConvertReductionOp;
3778 : public GenericConvertReductionOp<ConvertProdOp, TF::ProdOp, MulOp> {
[all …]
|