Class TensorPool

gpu.hpp:190–195

* @brief Represents a pool of tensors to manage GPU resources. The pool is * responsible for managing the lifetime of the tensors and freeing them when * the pool is destroyed. * * Most users do not need to interact with the TensorPool type, as there is a * member instance in the Context struct to simplify lifetime management of GPU * resources. */

 * resources.
 */
struct TensorPool {
  inline TensorPool(Context *ctx) : ctx(ctx), data() {};
  Context *ctx;
  std::unordered_map<WGPUBuffer, Tensor> data;
  ~TensorPool();
};

enum NumType {
  kf16, // (experimental)