auto shape_const_op = llvm::dyn_cast<TF::ConstOp>(shape_op);
  if (!shape_const_op) return failure();
  for (const auto& v : shape_const_op.getValue().getValues<APInt>()) {
    int64_t dim_size = v.getSExtValue();
    if (dim_size == tensorflow::kTFDynamicSize) return failure();
    shape->push_back(dim_size);
  }
  return success();
}

// Checks the result Variant type to infer the element shape if fully defined.

      auto shape_value_array = shape_value_attr.getValues<llvm::APInt>();
      for (int32_t idx = 0; idx < shape_value_array.size(); ++idx) {
        auto size = shape_value_array[idx].getSExtValue();
        new_shape_array_i32.push_back(
            ShapedType::isDynamic(size) ? -1 : static_cast<int32_t>(size));
      }
      return rewriter.create<arith::ConstantOp>(
          loc, DenseIntElementsAttr::get(