loc, DenseElementsAttr::get(tensorflow::GetTypeFromTFTensorShape(
                                        {}, rewriter.getIntegerType(64)),
                                    op.getAxis()));
    int64_t axis = op.getAxis();

    Type prev_input_ty, inferred_ty;
    SmallVector<Value, 4> expanded_inputs;
    expanded_inputs.reserve(op.getN());
    for (Value input : op.getValues()) {