#define GEN_PASS_DEF_PUSHTRANSPOSETHROUGHEWISEPASS
#include "tensorflow/compiler/mlir/lite/transforms/passes.h.inc"

class PushTransposeThroughEwisePass
    : public impl::PushTransposeThroughEwisePassBase<
          PushTransposeThroughEwisePass> {
 public:
  MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(PushTransposeThroughEwisePass)
  void runOnOperation() override;
};

// Compute the permutation that would take `arr` to the identity.

  let summary = "Reduce some tensor types' precision if all values are within range.";
  let constructor = "CreateReduceTypePrecisionPass()";
  let dependentDialects = ["TFL::TensorFlowLiteDialect"];
}

def PushTransposeThroughEwisePass : Pass<"push-transpose-through-ewise", "mlir::ModuleOp"> {
  let summary = "TODO";
  let constructor = "CreatePushTransposeThroughEwisePass()";
  let dependentDialects = ["TFL::TensorFlowLiteDialect"];