ML functions
 
Loading...
Searching...
No Matches
TorchDNNV2CUDA Member List

This is the complete list of members for TorchDNNV2CUDA, including all inherited members.

apply(const SelectivityVector &rows, std::vector< VectorPtr > &args, const TypePtr &type, exec::EvalCtx &context, VectorPtr &output) const overrideTorchDNNV2CUDAinline
dimsMLFunctionprotected
getBias() constTorchDNNV2CUDAinline
getCoefficientVector(std::string name)MLFunctioninlineprotected
getCost(std::vector< int > inputDims)TorchDNNV2CUDAinlinevirtual
getDims()MLFunctioninlinevirtual
getFuncName()TorchDNNV2CUDAinlinevirtual
getKernelTypes() constTorchDNNV2CUDAinline
getName()TorchDNNV2CUDAinlinestatic
getNumDims()MLFunctioninlinevirtual
getTensor() const overrideTorchDNNV2CUDAinlinevirtual
getWeightedCost(std::string name, float cost)MLFunctioninlineprotected
getWeights() constTorchDNNV2CUDAinline
signatures()TorchDNNV2CUDAinlinestatic
TorchDNNV2CUDA(std::vector< velox::dl::KernelType > kernelTypes, std::vector< float * > weights, std::vector< int > dimensions)TorchDNNV2CUDAinline
~MLFunction()=defaultMLFunctionvirtual