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, bool initialize_parameters=true)	Mila::Dnn::BuildContext	inlineexplicit
getPrefillSize() const noexcept	Mila::Dnn::BuildContext	inline
getRuntimeMode() const noexcept	Mila::Dnn::BuildContext	inline
initialize_parameters_	Mila::Dnn::BuildContext	private
input_shape_	Mila::Dnn::BuildContext	private
inputShape() const noexcept	Mila::Dnn::BuildContext	inline
isInferenceMode() const noexcept	Mila::Dnn::BuildContext	inline
isTrainingMode() const noexcept	Mila::Dnn::BuildContext	inline
prefill_size_	Mila::Dnn::BuildContext	private
runtime_mode_	Mila::Dnn::BuildContext	private
shouldInitializeParameters() const noexcept	Mila::Dnn::BuildContext	inline
withPrefillSize(int64_t prefill_size) const	Mila::Dnn::BuildContext	inline
withShape(shape_t new_shape) const	Mila::Dnn::BuildContext	inline