Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Compute::Cuda::MatMulBiasGelu::CudaMatMulBiasGeluOp< TInput, TOutput > Member List

This is the complete list of members for Mila::Dnn::Compute::Cuda::MatMulBiasGelu::CudaMatMulBiasGeluOp< TInput, TOutput >, including all inherited members.

asInputTensor(const ITensor &t)Mila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, float, float >inlineprotectedstatic
asOutputTensor(ITensor &t)Mila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, float, float >inlineprotectedstatic
backward(const Tensor< TInput, MR > &input, const Tensor< TOutput, MR > &output, const Tensor< TOutput, MR > &output_gradient, const std::vector< std::shared_ptr< ITensor > > &parameters, std::vector< std::shared_ptr< Tensor< TOutput, MR > > > &parameter_gradients, Tensor< TInput, MR > &input_gradient, const std::vector< std::shared_ptr< Tensor< TOutput, MR > > > &output_state) constMila::Dnn::Compute::Cuda::MatMulBiasGelu::CudaMatMulBiasGeluOp< TInput, TOutput >inline
Mila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, float, float >::backward(const ITensor &input, const ITensor &output_grad, ITensor &input_grad) const=0Mila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, float, float >pure virtual
build(const BuildContext &build_context)Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >inlinevirtual
className()Mila::Dnn::Compute::Cuda::MatMulBiasGelu::CudaMatMulBiasGeluOp< TInput, TOutput >inlinestatic
clearGradients() noexceptMila::Dnn::Compute::Operation< TDeviceType, TPrecision >inlinevirtual
CudaMatMulBiasGeluOp()Mila::Dnn::Compute::Cuda::MatMulBiasGelu::CudaMatMulBiasGeluOp< TInput, TOutput >inline
data_typeMila::Dnn::Compute::Operation< TDeviceType, TPrecision >static
DataTypeTraits typedefMila::Dnn::Compute::Operation< TDeviceType, TPrecision >
device_typeMila::Dnn::Compute::Operation< TDeviceType, TPrecision >static
forward(const Tensor< TInput, MR > &input, const std::vector< std::shared_ptr< ITensor > > &parameters, Tensor< TOutput, MR > &output, std::vector< std::shared_ptr< Tensor< TOutput, MR > > > &output_state) const overrideMila::Dnn::Compute::Cuda::MatMulBiasGelu::CudaMatMulBiasGeluOp< TInput, TOutput >inline
Mila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, float, float >::forward(const ITensor &input, ITensor &output) const=0Mila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, float, float >pure virtual
getDataType() constMila::Dnn::Compute::Operation< TDeviceType, TPrecision >inlinevirtual
getDeviceType() constMila::Dnn::Compute::Operation< TDeviceType, TPrecision >inlinevirtual
getName() const overrideMila::Dnn::Compute::Cuda::MatMulBiasGelu::CudaMatMulBiasGeluOp< TInput, TOutput >inlinevirtual
getOperationType() const=0Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >pure virtual
getStateMemorySize() constMila::Dnn::Compute::Operation< TDeviceType, TPrecision >inlinevirtual
is_built_Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >protected
isBuilt() constMila::Dnn::Compute::Operation< TDeviceType, TPrecision >inlinevirtual
isEvalMode() constMila::Dnn::Compute::Operation< TDeviceType, TPrecision >inlinevirtual
MR typedefMila::Dnn::Compute::Cuda::MatMulBiasGelu::CudaMatMulBiasGeluOp< TInput, TOutput >
setGradients(ITensor *weight_grad, ITensor *bias_grad)Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >inlinevirtual
setParameters(ITensor *weight, ITensor *bias)Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >inlinevirtual
setTrainingMode(TrainingMode training_mode)Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >inlinevirtual
TensorInputType typedefMila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, float, float >
TensorOutputType typedefMila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, float, float >
training_mode_Mila::Dnn::Compute::Operation< TDeviceType, TPrecision >protected
UnaryOperationBase typedefMila::Dnn::Compute::Cuda::MatMulBiasGelu::CudaMatMulBiasGeluOp< TInput, TOutput >
~Operation()=defaultMila::Dnn::Compute::Operation< TDeviceType, TPrecision >virtual
~UnaryOperation()=defaultMila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, float, float >virtual