Mila
Deep Neural Network Library
|
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 > > > ¶meter_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::CpuLinearOp | inline |
UnaryOperation< DeviceType::Cpu, float >::backward(const Tensor< float, MR > &grad, const std::vector< std::shared_ptr< Tensor< float, MR > > > ¶meters, std::vector< std::shared_ptr< Tensor< float, MR > > > &output_grads) const | Mila::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 > > > ¶meters, std::vector< std::shared_ptr< Tensor< float, MR > > > ¶meter_grads, Tensor< float, MR > &input_grad, const OperationAttributes &properties, const std::vector< std::shared_ptr< Tensor< float, MR > > > &output_state) const | Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, float > | inlinevirtual |
config_ | Mila::Dnn::Compute::CpuLinearOp | private |
CpuLinearOp(const LinearConfig &config) | Mila::Dnn::Compute::CpuLinearOp | inline |
CpuLinearOp(std::shared_ptr< DeviceContext > context, const LinearConfig &config) | Mila::Dnn::Compute::CpuLinearOp | inline |
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 > > > ¶meters, const OperationAttributes &properties, Tensor< float, MR > &output, std::vector< std::shared_ptr< Tensor< float, MR > > > &output_state) const override | Mila::Dnn::Compute::CpuLinearOp | inline |
UnaryOperation< DeviceType::Cpu, float >::forward(const Tensor< float, MR > &input, const std::vector< std::shared_ptr< Tensor< float, MR > > > ¶meters, const OperationAttributes &properties, Tensor< float, MR > &output, std::vector< std::shared_ptr< Tensor< float, MR > > > &output_state) const=0 | Mila::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) const | Mila::Dnn::Compute::CpuLinearOp | inlineprivate |
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::CpuLinearOp | inlinevirtual |
getOperationType() const | Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput > | inline |
MR typedef | Mila::Dnn::Compute::CpuLinearOp | |
operation_type_ | Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput > | private |
OperationBase typedef | Mila::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()=default | Mila::Dnn::Compute::OperationBase< TDeviceType, TInput1, TInput2, TOutput > | virtual |
~UnaryOperation()=default | Mila::Dnn::Compute::UnaryOperation< DeviceType::Cpu, float > | virtual |