Method setShape

src/backend/cpu/jit/BufferNode.hpp:61–71

    }

    void setShape(af::dim4 new_shape) final {
        auto new_strides = calcStrides(new_shape);
        m_dims[0]        = new_shape[0];
        m_dims[1]        = new_shape[1];
        m_dims[2]        = new_shape[2];
        m_dims[3]        = new_shape[3];
        m_strides[0]     = new_strides[0];
        m_strides[1]     = new_strides[1];
        m_strides[2]     = new_strides[2];
        m_strides[3]     = new_strides[3];
    }

    void calc(int x, int y, int z, int w, int lim) final {
        using Tc = compute_t<T>;