Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Compute::CpuLinearOp Member List

This is the complete list of members for Mila::Dnn::Compute::CpuLinearOp, including all inherited members.

backward(Tensor< float, MR > &input_grad, const std::vector< std::shared_ptr< Tensor< float, MR > > > &parameter_grads, const Tensor< float, MR > &output_grad, const Tensor< float, MR > input, const Tensor< float, MR > weight, int B, int T, int C, int OC)Mila::Dnn::Compute::CpuLinearOpinline
UnaryOperation< DeviceType::Cpu, float >::backward(const Tensor< float, MR > &grad, const std::vector< std::shared_ptr< Tensor< float, MR > > > &parameters, std::vector< std::shared_ptr< Tensor< float, MR > > > &output_grads) constMila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, float >inlinevirtual
UnaryOperation< DeviceType::Cpu, float >::backward(const Tensor< float, MR > &input, const Tensor< float, MR > &output_grad, const std::vector< std::shared_ptr< Tensor< float, MR > > > &parameters, std::vector< std::shared_ptr< Tensor< float, MR > > > &parameter_grads, Tensor< float, MR > &input_grad, const OperationAttributes &properties, const std::vector< std::shared_ptr< Tensor< float, MR > > > &output_state) constMila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, float >inlinevirtual
config_Mila::Dnn::Compute::CpuLinearOpprivate
CpuLinearOp(const LinearConfig &config)Mila::Dnn::Compute::CpuLinearOpinline
CpuLinearOp(std::shared_ptr< DeviceContext > context, const LinearConfig &config)Mila::Dnn::Compute::CpuLinearOpinline
device_context_Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >private
forward(const Tensor< float, MR > &input, const std::vector< std::shared_ptr< Tensor< float, MR > > > &parameters, const OperationAttributes &properties, Tensor< float, MR > &output, std::vector< std::shared_ptr< Tensor< float, MR > > > &output_state) const overrideMila::Dnn::Compute::CpuLinearOpinline
UnaryOperation< DeviceType::Cpu, float >::forward(const Tensor< float, MR > &input, const std::vector< std::shared_ptr< Tensor< float, MR > > > &parameters, const OperationAttributes &properties, Tensor< float, MR > &output, std::vector< std::shared_ptr< Tensor< float, MR > > > &output_state) const=0Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, float >pure virtual
forward_naive(const Tensor< float, MR > &input, const std::shared_ptr< Tensor< float, MR > > &weight, const std::shared_ptr< Tensor< float, MR > > &bias, Tensor< float, MR > &output, int outer_size, int C, int OC) constMila::Dnn::Compute::CpuLinearOpinlineprivate
getDeviceContext() constMila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >inline
getDeviceType() constMila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >inline
getName() const overrideMila::Dnn::Compute::CpuLinearOpinlinevirtual
getOperationType() constMila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >inline
MR typedefMila::Dnn::Compute::CpuLinearOp
operation_type_Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >private
OperationBase typedefMila::Dnn::Compute::CpuLinearOp
UnaryOperation< DeviceType::Cpu, float >::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::Cpu, float >inline
UnaryOperation(OperationType operation_type, std::shared_ptr< DeviceContext > context)Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, float >inline
~OperationBase()=defaultMila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput >virtual
~UnaryOperation()=defaultMila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, float >virtual