Trait std::DeviceTensor

#include <std/ml/tensor.h> source

trait DeviceTensor : Tensor {
    int device() const;
    int place(int device);
    int sync();
}

What a tensor that can live on a device adds (device.h): DenseTensor and SparseTensor (sparse.h) conform. device() where the values live (DEVICE_CPU, DEVICE_ROCM, DEVICE_CUDA, DEVICE_MPS) place(d) moves the tensor itself to d (parameters, leaves); 1 on success sync() makes the host copy current; 1 on success

Required methods

§int device() constsource

§int place(int device)source

§int sync()source