Mila::Dnn::BuildContext Member List

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

BuildContext()	Mila::Dnn::BuildContext	inline
BuildContext(shape_t input_shape, RuntimeMode runtime_mode, std::optional< bool > initialize_parameters=std::nullopt)	Mila::Dnn::BuildContext	inlineexplicit
getPrefillSize() const noexcept	Mila::Dnn::BuildContext	inline
getRuntimeMode() const noexcept	Mila::Dnn::BuildContext	inline
hasInstalledOutput() const noexcept	Mila::Dnn::BuildContext	inline
inputShape() const noexcept	Mila::Dnn::BuildContext	inline
isInferenceMode() const noexcept	Mila::Dnn::BuildContext	inline
isTrainingMode() const noexcept	Mila::Dnn::BuildContext	inline
shouldInitializeParameters() const noexcept	Mila::Dnn::BuildContext	inline
withInstalledOutput(bool installed) const	Mila::Dnn::BuildContext	inline
withPrefillSize(int64_t prefill_size) const	Mila::Dnn::BuildContext	inline
withShape(shape_t new_shape) const	Mila::Dnn::BuildContext	inline