Method size

mlx/data/Buffer.cpp:25–27

  return self_->get(idx);
}
int64_t Buffer::size() const {
  return self_->size();
}

Buffer Buffer::batch(
    int64_t batch_size,