Method size

csrc/adam/cpu_adam_impl.cpp:400–400

    }

    size_t size() const { return n_; }

    // Split [0, total) into one chunk per thread. Chunk boundaries are rounded up to a
    // multiple of `align` so each slice's AVX/scalar split lines up with the whole-tensor