NEML2 2.0.0
|
This is the complete list of members for TensorLoader, including all inherited members.
generate(std::size_t n) override | TensorLoader | protectedvirtual |
has_more() const override | FixedSizeWorkGenerator< Tensor > | inlinevirtual |
next(std::size_t n) | WorkGenerator< Tensor > | inline |
offset() const | WorkGenerator< Tensor > | inline |
operator=(WorkGenerator &&) noexcept=default | WorkGenerator< Tensor > | |
operator=(const WorkGenerator &)=default | WorkGenerator< Tensor > | |
TensorLoader(const Tensor &tensor, Size batch_dim) | TensorLoader | |
total() const override | TensorLoader | virtual |
WorkGenerator()=default | WorkGenerator< Tensor > | |
WorkGenerator(WorkGenerator &&) noexcept=default | WorkGenerator< Tensor > | |
WorkGenerator(const WorkGenerator &)=default | WorkGenerator< Tensor > | |
~WorkGenerator()=default | WorkGenerator< Tensor > | virtual |