| backward(const Tensor< TInput, MR > &input, const Tensor< TOutput, MR > &output_grad, Tensor< TInput, MR > &input_grad) | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | inline |
| build(const BuildContext &context) final | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlinevirtual |
| build_context_ | Mila::Dnn::Component< DeviceType::Cuda, float, float > | protected |
| built_ | Mila::Dnn::Component< DeviceType::Cuda, float, float > | private |
| Component(const std::string &name) | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlineexplicit |
| ComponentBase typedef | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | |
| config_ | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | private |
| createOperation() | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | inlineprivate |
| Dropout(const std::string &device_name, const DropoutConfig &config) | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | inlineexplicit |
| Dropout(std::shared_ptr< DeviceContext > device_context, const DropoutConfig &config) | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | inlineexplicit |
| ensureBuilt(const char *method) const | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlineprivate |
| exec_context_ | Mila::Dnn::Component< DeviceType::Cuda, float, float > | private |
| forward(const Tensor< TInput, MR > &input, Tensor< TOutput, MR > &output) | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | inline |
| generateMask(Tensor< TOutput, MR > &mask, const std::vector< int64_t > &shape) | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | inlineprivate |
| getDeviceId() const=0 | Mila::Dnn::Component< DeviceType::Cuda, float, float > | pure virtual |
| getDeviceType() | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlinestatic |
| getExecutionContext() const | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlineprotected |
| getGradients() const=0 | Mila::Dnn::Component< DeviceType::Cuda, float, float > | pure virtual |
| getMemoryStats() const=0 | Mila::Dnn::Component< DeviceType::Cuda, float, float > | pure virtual |
| getName() const | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inline |
| getParameterNames() const | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlinevirtual |
| getParameters() const=0 | Mila::Dnn::Component< DeviceType::Cuda, float, float > | pure virtual |
| getPrecision() noexcept | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlinestatic |
| getProbability() const | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | inline |
| getRuntimeMode() const noexcept | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inline |
| getSeed() const | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | inline |
| getTrainingMode() const noexcept | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inline |
| getType() const override | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | inlinevirtual |
| hasExecutionContext() const noexcept | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlineprotected |
| isBuilt() const final | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlinevirtual |
| isIdentifier(const std::string &s) noexcept | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlineprivatestatic |
| isInferenceMode() const noexcept | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inline |
| isTrainingMode() const noexcept | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inline |
| load(ModelArchive &archive) override | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | inline |
| loadParameter(const std::string &name, const Serialization::ITensorBlob &blob) | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlinevirtual |
| loadParameterFromBlob(const std::string ¶m_name, const Serialization::ITensorBlob &blob, Tensor< TParameterPrecision, TMemoryResource > &target, const shape_t &expected_shape) | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlineprotected |
| mask_ | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | private |
| MR typedef | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | |
| name_ | Mila::Dnn::Component< DeviceType::Cuda, float, float > | private |
| onBuilding(const BuildContext &config) | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlineprotectedvirtual |
| onExecutionContextSet() | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlineprotectedvirtual |
| onTrainingModeChanging(TrainingMode mode) | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlineprotectedvirtual |
| operation_ | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | private |
| output_state_ | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | private |
| parameterCount() const override | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | inlinevirtual |
| parameters_ | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | private |
| rng_ | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | private |
| save(ModelArchive &archive) const override | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | inline |
| save_(ModelArchive &archive, SerializationMode mode) const=0 | Mila::Dnn::Component< DeviceType::Cuda, float, float > | pure virtual |
| seed_ | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | private |
| setExecutionContext(IExecutionContext *context) | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlineprotected |
| setSeed(unsigned int seed) | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | inline |
| setTrainingMode(TrainingMode mode) | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inline |
| synchronize()=0 | Mila::Dnn::Component< DeviceType::Cuda, float, float > | pure virtual |
| toString() const override | Mila::Dnn::Dropout< TDeviceType, TInput, TOutput > | inlinevirtual |
| training_mode_ | Mila::Dnn::Component< DeviceType::Cuda, float, float > | private |
| training_mode_mutex_ | Mila::Dnn::Component< DeviceType::Cuda, float, float > | private |
| validateName(const std::string &name) | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlineprivatestatic |
| zeroGradients() | Mila::Dnn::Component< DeviceType::Cuda, float, float > | inlinevirtual |
| ~Component()=default | Mila::Dnn::Component< DeviceType::Cuda, float, float > | virtual |
1.15.0