Scanner C++ API
Public Member Functions | Friends | List of all members
scanner::internal::KernelBuilder Class Reference

Public Member Functions

 KernelBuilder (const std::string &name, KernelConstructor constructor)
 
KernelBuilder & device (DeviceType device_type)
 
KernelBuilder & num_devices (i32 devices)
 
KernelBuilder & input_device (const std::string &input_name, DeviceType device_type)
 
KernelBuilder & output_device (const std::string &output_name, DeviceType device_type)
 
KernelBuilder & batch (i32 preferred_batch_size=1)
 

Friends

class KernelRegistration
 

The documentation for this class was generated from the following file: