load_driver.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_LOAD_DRIVER_H
#define FREE_TENSOR_LOAD_DRIVER_H
 
#include <driver/array.h>
#include <driver/device.h>
#include <driver/target.h>
#include <ref.h>
namespace freetensor {
 
Ref<Target> loadTarget(const std::string &txt, const std::string &data);
Ref<Device> loadDevice(const std::string &txt, const std::string &data);
Ref<Array> loadArray(const std::string &txt, const std::string &data);
 
Ref<Array> newArray(const std::vector<size_t> &shape_,
                    const std::string &dtype_, const std::string &data_);
 
} // namespace freetensor
 
#endif // FREE_TENSOR_LOAD_DRIVER_H