Mila::Dnn::Compute::CudaGeluOp< TDataType > Member List

This is the complete list of members for Mila::Dnn::Compute::CudaGeluOp< TDataType >, including all inherited members.

backward(const Tensor< TDataType, MR > &input, const Tensor< TDataType, MR > &output, const Tensor< TDataType, MR > &output_gradient, const std::vector< std::shared_ptr< Tensor< TDataType, MR > > > &parameters, std::vector< std::shared_ptr< Tensor< TDataType, MR > > > &parameter_gradients, Tensor< TDataType, MR > &input_gradient, const OperationAttributes &properties, const std::vector< std::shared_ptr< Tensor< TDataType, MR > > > &output_state) const	Mila::Dnn::Compute::CudaGeluOp< TDataType >	inline
UnaryOperation< DeviceType::Cuda, TDataType, TDataType >::backward(const Tensor< TDataType, MR > &grad, const std::vector< std::shared_ptr< Tensor< TDataType, MR > > > &parameters, std::vector< std::shared_ptr< Tensor< TDataType, MR > > > &output_grads) const	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, TDataType, TDataType >	inlinevirtual
UnaryOperation< DeviceType::Cuda, TDataType, TDataType >::backward(const Tensor< TDataType, MR > &input, const Tensor< TDataType, MR > &output_grad, const std::vector< std::shared_ptr< Tensor< TDataType, MR > > > &parameters, std::vector< std::shared_ptr< Tensor< TDataType, MR > > > &parameter_grads, Tensor< TDataType, MR > &input_grad, const OperationAttributes &properties, const std::vector< std::shared_ptr< Tensor< TDataType, MR > > > &output_state) const	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, TDataType, TDataType >	inlinevirtual
config_	Mila::Dnn::Compute::CudaGeluOp< TDataType >	private
CudaGeluOp(const GeluConfig &config)	Mila::Dnn::Compute::CudaGeluOp< TDataType >	inline
CudaGeluOp(std::shared_ptr< DeviceContext > context, const GeluConfig &config)	Mila::Dnn::Compute::CudaGeluOp< TDataType >	inline
device_context_	Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >	private
forward(const Tensor< TDataType, MR > &input, const std::vector< std::shared_ptr< Tensor< TDataType, MR > > > &parameters, const OperationAttributes &properties, Tensor< TDataType, MR > &output, std::vector< std::shared_ptr< Tensor< TDataType, MR > > > &output_state) const override	Mila::Dnn::Compute::CudaGeluOp< TDataType >	inline
UnaryOperation< DeviceType::Cuda, TDataType, TDataType >::forward(const Tensor< TDataType, MR > &input, const std::vector< std::shared_ptr< Tensor< TDataType, MR > > > &parameters, const OperationAttributes &properties, Tensor< TDataType, MR > &output, std::vector< std::shared_ptr< Tensor< TDataType, MR > > > &output_state) const=0	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, TDataType, TDataType >	pure virtual
getConfig() const	Mila::Dnn::Compute::CudaGeluOp< TDataType >	inline
getDeviceContext() const	Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >	inline
getDeviceType() const	Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >	inline
getName() const override	Mila::Dnn::Compute::CudaGeluOp< TDataType >	inlinevirtual
getOperationType() const	Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >	inline
impl_	Mila::Dnn::Compute::CudaGeluOp< TDataType >	private
MR typedef	Mila::Dnn::Compute::CudaGeluOp< TDataType >
operation_type_	Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >	private
OperationBase(OperationType operation_type, std::shared_ptr< DeviceContext > context)	Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >	inline
UnaryOperation(OperationType operation_type)	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, TDataType, TDataType >	inline
UnaryOperation(OperationType operation_type, std::shared_ptr< DeviceContext > context)	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, TDataType, TDataType >	inline
UnaryOperationBase typedef	Mila::Dnn::Compute::CudaGeluOp< TDataType >
~OperationBase()=default	Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >	virtual
~UnaryOperation()=default	Mila::Dnn::Compute::UnaryOperation< DeviceType::Cuda, TDataType, TDataType >	virtual