PatternRewriter &rewriter) const override {
    auto result_value = gather_nd_first.getResult();
    auto params_value = gather_nd_first.getParams();
    auto indices_value = gather_nd_first.getIndices();

    auto result_type = dyn_cast<RankedTensorType>(result_value.getType());
    auto params_type = dyn_cast<RankedTensorType>(params_value.getType());

  DenseIntElementsAttr axis_attr;
  if (matchPattern(op.getAxis(), m_Constant(&axis_attr))) {
    int64_t axis = (*axis_attr.begin()).getSExtValue();
    if (auto ty = mlir::dyn_cast<RankedTensorType>(op.getParams().getType())) {
      int64_t rank = ty.getRank();
      if (axis >= rank || axis < -rank)
        return op.emitOpError() << "axis (" << axis << ") must be in range ["

    return req;
  }
  return {};
}

ValuePort ComputeInputComponentFor(GatherV2Op op, ArrayRef<unsigned int> port) {
  if (port.size() != 2) return {};
  assert(port[0] == 0);

  auto params = op.getParams();
  auto params_ty = mlir::dyn_cast<RankedTensorType>(params.getType());
  if (!params_ty || !params_ty.hasStaticShape() || params_ty.getRank() != 1 ||
      op.getBatchDims() != 0) {
    return {};
  }

// Gather op
//===----------------------------------------------------------------------===//

LogicalResult GatherOp::verify() {
  GatherOp op = *this;
  ShapedType params_type = op.getParams().getType().cast<ShapedType>();
  // TFLite gather kernel supports 1D string input only.
  if (params_type.getElementType().isa<mlir::TF::StringType>()) {
    if (params_type.hasRank() && params_type.getRank() != 1) {

Method <org.gradle.api.ActionConfiguration.getParams()> does not have raw return type assignable to org.gradle.api.provider.Property in (ActionConfiguration.java:0)
Method <org.gradle.api.AntBuilder.getAnt()> does not have raw return type assignable to org.gradle.api.provider.Provider in (AntBuilder.java:0)
Method <org.gradle.api.AntBuilder.getLifecycleLogLevel()> does not have raw return type assignable to org.gradle.api.provider.Property in (AntBuilder.java:0)