Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant > Member List

This is the complete list of members for Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >, including all inherited members.

backward(const TensorType &input, const TensorType &output_grad)Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inline
bias_Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >private
bias_grad_Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >private
build(const BuildContext &context) finalMila::Dnn::Component< TDeviceType, TComputePrecision >inlinevirtual
build_context_Mila::Dnn::Component< TDeviceType, TComputePrecision >protected
built_Mila::Dnn::Component< TDeviceType, TComputePrecision >private
Component(const std::string &name)Mila::Dnn::Component< TDeviceType, TComputePrecision >inlineexplicit
ComponentBase typedefMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >
config_Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >private
createOperation()Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlineprivate
ensureBuilt(const char *method) constMila::Dnn::Component< TDeviceType, TComputePrecision >inlineprivate
exec_context_Mila::Dnn::Component< TDeviceType, TComputePrecision >private
forward(const TensorType &input)Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inline
getConfig() const noexceptMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inline
getDeviceId() const overrideMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlinevirtual
getDeviceType()Mila::Dnn::Component< TDeviceType, TComputePrecision >inlinestatic
getExecutionContext() constMila::Dnn::Component< TDeviceType, TComputePrecision >inlineprotected
getGradients() const overrideMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlinevirtual
getMemoryStats() const overrideMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlinevirtual
getName() constMila::Dnn::Component< TDeviceType, TComputePrecision >inline
getParameterNames() constMila::Dnn::Component< TDeviceType, TComputePrecision >inlinevirtual
getParameters() const overrideMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlinevirtual
getPrecision() noexceptMila::Dnn::Component< TDeviceType, TComputePrecision >inlinestatic
getRuntimeMode() const noexceptMila::Dnn::Component< TDeviceType, TComputePrecision >inline
getTrainingMode() const noexceptMila::Dnn::Component< TDeviceType, TComputePrecision >inline
getType() const overrideMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlinevirtual
hasBias() const noexceptMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inline
hasExecutionContext() const noexceptMila::Dnn::Component< TDeviceType, TComputePrecision >inlineprotected
initializeGradients()Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlineprivate
initializeParameters(const BuildContext &context)Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlineprivate
input_grad_Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >private
isBuilt() const finalMila::Dnn::Component< TDeviceType, TComputePrecision >inlinevirtual
isIdentifier(const std::string &s) noexceptMila::Dnn::Component< TDeviceType, TComputePrecision >inlineprivatestatic
isInferenceMode() const noexceptMila::Dnn::Component< TDeviceType, TComputePrecision >inline
isTrainingMode() const noexceptMila::Dnn::Component< TDeviceType, TComputePrecision >inline
kIsQuantizedMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >static
kWeightDtypeMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >static
leading_shape_Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >private
Linear(const std::string &name, const LinearConfig &config, std::optional< DeviceId > device_id=std::nullopt)Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlineexplicit
loadParameter(const std::string &name, const ITensorBlob &blob) overrideMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlinevirtual
loadParameterFromBlob(const std::string &param_name, const Serialization::ITensorBlob &blob, Tensor< TParameterPrecision, TMemoryResource > &target, const shape_t &expected_shape)Mila::Dnn::Component< TDeviceType, TComputePrecision >inlineprotected
MR typedefMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >
name_Mila::Dnn::Component< TDeviceType, TComputePrecision >private
onBuilding(const BuildContext &context) overrideMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlineprotectedvirtual
onExecutionContextSet() overrideMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlineprotectedvirtual
onTrainingModeChanging(TrainingMode mode) overrideMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlineprotectedvirtual
operation_Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >private
OpType typedefMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >
output_Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >private
output_view_Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >private
owned_exec_context_Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >private
parameterCount() const overrideMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlinevirtual
save_(ModelArchive &archive, SerializationMode mode) const overrideMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlinevirtual
setExecutionContext(IExecutionContext *context)Mila::Dnn::Component< TDeviceType, TComputePrecision >inlineprotected
setTrainingMode(TrainingMode mode)Mila::Dnn::Component< TDeviceType, TComputePrecision >inline
synchronize() overrideMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlinevirtual
TensorType typedefMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >
toString() const overrideMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlinevirtual
training_mode_Mila::Dnn::Component< TDeviceType, TComputePrecision >private
training_mode_mutex_Mila::Dnn::Component< TDeviceType, TComputePrecision >private
validateBuildContext(const BuildContext &context) constMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlineprivate
validateInputShape(const shape_t &input_shape) constMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlineprivate
validateName(const std::string &name)Mila::Dnn::Component< TDeviceType, TComputePrecision >inlineprivatestatic
weight_Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >private
weight_grad_Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >private
weight_scales_Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >private
WeightScaleTensorType typedefMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >
WeightTensorType typedefMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >
zeroGradients() overrideMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >inlinevirtual
~Component()=defaultMila::Dnn::Component< TDeviceType, TComputePrecision >virtual
~Linear() override=defaultMila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >