Index

A

abs_binary_sign_grad() (in module brevitas.function.ops_ste) abs_binary_sign_grad_impl() (in module brevitas.ops.autograd_ste_ops) AbsAve (class in brevitas.core.stats.stats_op) AbsBinarySignGradFn (class in brevitas.ops.autograd_ste_ops)

AbsMax (class in brevitas.core.stats.stats_op) AbsMaxAve (class in brevitas.core.stats.stats_op) AbsMaxL2 (class in brevitas.core.stats.stats_op) AbsMinMax (class in brevitas.core.stats.stats_op) AbsPercentile (class in brevitas.core.stats.stats_op)

B

binary_sign() (in module brevitas.function.ops) binary_sign_ste() (in module brevitas.function.ops_ste) binary_sign_ste_impl() (in module brevitas.ops.autograd_ste_ops) BinaryQuant (class in brevitas.core.quant.binary) BinarySignSteFn (class in brevitas.ops.autograd_ste_ops) BitWidthConst (class in brevitas.core.bit_width.const) BitWidthParameter (class in brevitas.core.bit_width.parameter) BitWidthStatefulConst (class in brevitas.core.bit_width.const) brevitas.core module brevitas.core.bit_width module brevitas.core.bit_width.const module brevitas.core.bit_width.parameter module brevitas.core.function_wrapper module brevitas.core.function_wrapper.clamp module brevitas.core.function_wrapper.misc module brevitas.core.function_wrapper.ops_ste module brevitas.core.function_wrapper.shape module brevitas.core.quant module brevitas.core.quant.binary module brevitas.core.quant.delay module brevitas.core.quant.int module

brevitas.core.quant.int_base module brevitas.core.quant.ternary module brevitas.core.restrict_val module brevitas.core.scaling module brevitas.core.scaling.int_scaling module brevitas.core.scaling.runtime module brevitas.core.scaling.standalone module brevitas.core.stats module brevitas.core.stats.stats_op module brevitas.core.stats.stats_wrapper module brevitas.core.stats.view_wrapper module brevitas.core.utils module brevitas.function.ops module brevitas.function.ops_ste module brevitas.function.shape module brevitas.ops.autograd_ste_ops module

C

ceil_ste() (in module brevitas.function.ops_ste) ceil_ste_impl() (in module brevitas.ops.autograd_ste_ops) CeilSte (class in brevitas.core.function_wrapper.ops_ste) CeilSteFn (class in brevitas.ops.autograd_ste_ops)

ClampedBinaryQuant (class in brevitas.core.quant.binary) ClampMin (class in brevitas.core.function_wrapper.clamp) compute_max_mantissa() (in module brevitas.function.ops) ConstScaling (class in brevitas.core.scaling.standalone)

D

DecoupledIntQuant (class in brevitas.core.quant.int_base) DecoupledRescalingIntQuant (class in brevitas.core.quant.int) DecoupledRescalingIntQuantWithInput (class in brevitas.core.quant.int) DelayWrapper (class in brevitas.core.quant.delay) dpu_round() (in module brevitas.function.ops)

dpu_round_ste() (in module brevitas.function.ops_ste) dpu_round_ste_impl() (in module brevitas.ops.autograd_ste_ops) DPURoundSte (class in brevitas.core.function_wrapper.ops_ste) DPURoundSteFn (class in brevitas.ops.autograd_ste_ops) DYNAMIC_OVER_SUBCHANNEL_BLOCK (brevitas.core.function_wrapper.shape.StatsInputViewShapeImpl attribute) DynamicOverSubChannelBlockView (class in brevitas.core.function_wrapper.shape)

E

eager_forward() (brevitas.core.utils.SliceTensor method)

evaluate_loss() (brevitas.core.stats.stats_op.MSE method)

F

FloatClamp (class in brevitas.core.function_wrapper.clamp) FloatRestrictValue (class in brevitas.core.restrict_val) floor_ste() (in module brevitas.function.ops_ste) floor_ste_impl() (in module brevitas.ops.autograd_ste_ops) FloorSte (class in brevitas.core.function_wrapper.ops_ste) FloorSteFn (class in brevitas.ops.autograd_ste_ops) forward() (brevitas.core.bit_width.const.BitWidthConst method) (brevitas.core.bit_width.const.BitWidthStatefulConst method) (brevitas.core.bit_width.const.MsbClampBitWidth method) (brevitas.core.bit_width.parameter.BitWidthParameter method) (brevitas.core.bit_width.parameter.RemoveBitwidthParameter method) (brevitas.core.function_wrapper.clamp.ClampMin method) (brevitas.core.function_wrapper.clamp.FloatClamp method) (brevitas.core.function_wrapper.clamp.ScalarClamp method) (brevitas.core.function_wrapper.clamp.TensorClamp method) (brevitas.core.function_wrapper.misc.Identity method) (brevitas.core.function_wrapper.misc.InplaceLogTwo method) (brevitas.core.function_wrapper.misc.LogTwo method) (brevitas.core.function_wrapper.misc.PowerOfTwo method) (brevitas.core.function_wrapper.ops_ste.CeilSte method) (brevitas.core.function_wrapper.ops_ste.DPURoundSte method) (brevitas.core.function_wrapper.ops_ste.FloorSte method) (brevitas.core.function_wrapper.ops_ste.InplaceTensorClampSte method) (brevitas.core.function_wrapper.ops_ste.RoundSte method) (brevitas.core.function_wrapper.ops_ste.RoundToZeroSte method) (brevitas.core.function_wrapper.ops_ste.ScalarClampMinSte method) (brevitas.core.function_wrapper.ops_ste.TensorClampSte method) (brevitas.core.function_wrapper.shape.DynamicOverSubChannelBlockView method) (brevitas.core.function_wrapper.shape.OverBatchOverOutputChannelView method) (brevitas.core.function_wrapper.shape.OverBatchOverTensorView method) (brevitas.core.function_wrapper.shape.OverOutputChannelView method) (brevitas.core.function_wrapper.shape.OverOutputFeaturesView method) (brevitas.core.function_wrapper.shape.OverSubChannelBlockView method) (brevitas.core.function_wrapper.shape.OverTensorView method) (brevitas.core.function_wrapper.shape.PermuteDims method) (brevitas.core.quant.binary.BinaryQuant method) (brevitas.core.quant.binary.ClampedBinaryQuant method) (brevitas.core.quant.delay.DelayWrapper method) (brevitas.core.quant.int.DecoupledRescalingIntQuant method) (brevitas.core.quant.int.DecoupledRescalingIntQuantWithInput method) (brevitas.core.quant.int.PrescaledRestrictIntQuant method) (brevitas.core.quant.int.PrescaledRestrictIntQuantWithInputBitWidth method) (brevitas.core.quant.int.RescalingIntQuant method) (brevitas.core.quant.int.TruncIntQuant method) (brevitas.core.quant.int_base.DecoupledIntQuant method) (brevitas.core.quant.int_base.IntQuant method) (brevitas.core.quant.ternary.TernaryQuant method) (brevitas.core.restrict_val.FloatRestrictValue method) (brevitas.core.restrict_val.IntRestrictValue method) (brevitas.core.restrict_val.LogFloatRestrictValue method) (brevitas.core.restrict_val.PowerOfTwoRestrictValue method) (brevitas.core.restrict_val.QuantRestrictValue method) (brevitas.core.scaling.int_scaling.IntScaling method) (brevitas.core.scaling.int_scaling.PowerOfTwoIntScaling method) (brevitas.core.scaling.runtime.RuntimeDynamicGroupStatsScaling method) (brevitas.core.scaling.runtime.RuntimeStatsScaling method) (brevitas.core.scaling.runtime.StatsFromParameterScaling method) (brevitas.core.scaling.standalone.ConstScaling method) (brevitas.core.scaling.standalone.ParameterFromRuntimeStatsScaling method) (brevitas.core.scaling.standalone.ParameterFromStatsFromParameterScaling method) (brevitas.core.scaling.standalone.ParameterScaling method) (brevitas.core.scaling.standalone.TruncMsbScaling method) (brevitas.core.scaling.standalone.TruncScalingWrapper method) (brevitas.core.stats.stats_op.AbsAve method) (brevitas.core.stats.stats_op.AbsMax method) (brevitas.core.stats.stats_op.AbsMaxAve method) (brevitas.core.stats.stats_op.AbsMaxL2 method) (brevitas.core.stats.stats_op.AbsMinMax method) (brevitas.core.stats.stats_op.AbsPercentile method) (brevitas.core.stats.stats_op.HalfQuadraticOptimizerScale method) (brevitas.core.stats.stats_op.HalfQuadraticOptimizerZeroPoint method) (brevitas.core.stats.stats_op.KLMinimizerThreshold method) (brevitas.core.stats.stats_op.L1Norm method) (brevitas.core.stats.stats_op.L2Norm method) (brevitas.core.stats.stats_op.MeanLearnedSigmaStd method) (brevitas.core.stats.stats_op.MeanSigmaStd method) (brevitas.core.stats.stats_op.MSE method) (brevitas.core.stats.stats_op.NegativeMinOrZero method) (brevitas.core.stats.stats_op.NegativePercentileOrZero method) (brevitas.core.stats.stats_op.PercentileInterval method) (brevitas.core.utils.ParameterWrapper method) (brevitas.core.utils.SingleArgStatelessBuffer method) (brevitas.core.utils.SliceTensor method) (brevitas.core.utils.StatelessBuffer method)

G

get_upper_bound_on_l1_norm() (in module brevitas.function.ops)

H

HalfQuadraticOptimizerScale (class in brevitas.core.stats.stats_op)

HalfQuadraticOptimizerZeroPoint (class in brevitas.core.stats.stats_op)

I

Identity (class in brevitas.core.function_wrapper.misc) identity() (in module brevitas.function.ops) inf_nan_clamp() (brevitas.core.function_wrapper.clamp.FloatClamp method) init_scale() (brevitas.core.scaling.standalone.ParameterFromRuntimeStatsScaling method) inplace_momentum_update() (in module brevitas.core.utils) inplace_tensor_add() (in module brevitas.core.utils)

inplace_tensor_mul() (in module brevitas.core.utils) InplaceLogTwo (class in brevitas.core.function_wrapper.misc) InplaceTensorClampSte (class in brevitas.core.function_wrapper.ops_ste) InplaceTensorClampSteFn (class in brevitas.ops.autograd_ste_ops) IntQuant (class in brevitas.core.quant.int_base) IntRestrictValue (class in brevitas.core.restrict_val) IntScaling (class in brevitas.core.scaling.int_scaling)

K

KLMinimizerThreshold (class in brevitas.core.stats.stats_op)

L

L1Norm (class in brevitas.core.stats.stats_op) L2Norm (class in brevitas.core.stats.stats_op)

LogFloatRestrictValue (class in brevitas.core.restrict_val) LogTwo (class in brevitas.core.function_wrapper.misc)

M

masked_median() (in module brevitas.core.stats.stats_op) max_float() (in module brevitas.function.ops) max_int() (brevitas.core.quant.int.TruncIntQuant method) (brevitas.core.quant.int_base.DecoupledIntQuant method) (brevitas.core.quant.int_base.IntQuant method) (in module brevitas.function.ops) MeanLearnedSigmaStd (class in brevitas.core.stats.stats_op) MeanSigmaStd (class in brevitas.core.stats.stats_op) min_int() (brevitas.core.quant.int.TruncIntQuant method) (brevitas.core.quant.int_base.DecoupledIntQuant method) (brevitas.core.quant.int_base.IntQuant method) (in module brevitas.function.ops) module brevitas.core brevitas.core.bit_width brevitas.core.bit_width.const brevitas.core.bit_width.parameter brevitas.core.function_wrapper brevitas.core.function_wrapper.clamp brevitas.core.function_wrapper.misc brevitas.core.function_wrapper.ops_ste brevitas.core.function_wrapper.shape brevitas.core.quant brevitas.core.quant.binary brevitas.core.quant.delay brevitas.core.quant.int brevitas.core.quant.int_base brevitas.core.quant.ternary brevitas.core.restrict_val brevitas.core.scaling brevitas.core.scaling.int_scaling brevitas.core.scaling.runtime brevitas.core.scaling.standalone brevitas.core.stats brevitas.core.stats.stats_op brevitas.core.stats.stats_wrapper brevitas.core.stats.view_wrapper brevitas.core.utils brevitas.function.ops brevitas.function.ops_ste brevitas.function.shape brevitas.ops.autograd_ste_ops

MsbClampBitWidth (class in brevitas.core.bit_width.const) MSE (class in brevitas.core.stats.stats_op) mse_fib_search() (brevitas.core.stats.stats_op.MSE method) mse_grid_search() (brevitas.core.stats.stats_op.MSE method) mse_loss_fn() (brevitas.core.stats.stats_op.MSE method) mse_search() (brevitas.core.stats.stats_op.MSE method)

N

NegativeMinOrZero (class in brevitas.core.stats.stats_op)

NegativePercentileOrZero (class in brevitas.core.stats.stats_op)

O

optimize() (brevitas.core.stats.stats_op.HalfQuadraticOptimizerScale method) (brevitas.core.stats.stats_op.HalfQuadraticOptimizerZeroPoint method) OVER_BATCH_OVER_OUTPUT_CHANNELS (brevitas.core.function_wrapper.shape.StatsInputViewShapeImpl attribute) over_batch_over_output_channels() (in module brevitas.function.shape) OVER_BATCH_OVER_TENSOR (brevitas.core.function_wrapper.shape.StatsInputViewShapeImpl attribute) over_batch_over_tensor() (in module brevitas.function.shape) OVER_OUTPUT_CHANNELS (brevitas.core.function_wrapper.shape.StatsInputViewShapeImpl attribute) over_output_channels() (in module brevitas.function.shape) OVER_OUTPUT_FEATURES (brevitas.core.function_wrapper.shape.StatsInputViewShapeImpl attribute)

over_output_features() (in module brevitas.function.shape) OVER_SUBCHANNEL_BLOCK (brevitas.core.function_wrapper.shape.StatsInputViewShapeImpl attribute) OVER_TENSOR (brevitas.core.function_wrapper.shape.StatsInputViewShapeImpl attribute) over_tensor() (in module brevitas.function.shape) OverBatchOverOutputChannelView (class in brevitas.core.function_wrapper.shape) OverBatchOverTensorView (class in brevitas.core.function_wrapper.shape) OverOutputChannelView (class in brevitas.core.function_wrapper.shape) OverOutputFeaturesView (class in brevitas.core.function_wrapper.shape) OverSubChannelBlockView (class in brevitas.core.function_wrapper.shape) OverTensorView (class in brevitas.core.function_wrapper.shape)

P

parameter_search() (brevitas.core.stats.stats_op.HalfQuadraticOptimizerScale method) (brevitas.core.stats.stats_op.HalfQuadraticOptimizerZeroPoint method) ParameterFromRuntimeStatsScaling (class in brevitas.core.scaling.standalone) ParameterFromStatsFromParameterScaling (class in brevitas.core.scaling.standalone) ParameterScaling (class in brevitas.core.scaling.standalone) ParameterWrapper (class in brevitas.core.utils)

PercentileInterval (class in brevitas.core.stats.stats_op) PermuteDims (class in brevitas.core.function_wrapper.shape) PowerOfTwo (class in brevitas.core.function_wrapper.misc) PowerOfTwoIntScaling (class in brevitas.core.scaling.int_scaling) PowerOfTwoRestrictValue (class in brevitas.core.restrict_val) PrescaledRestrictIntQuant (class in brevitas.core.quant.int) PrescaledRestrictIntQuantWithInputBitWidth (class in brevitas.core.quant.int)

Q

QuantRestrictValue (class in brevitas.core.restrict_val)

R

RemoveBitwidthParameter (class in brevitas.core.bit_width.parameter) RescalingIntQuant (class in brevitas.core.quant.int) restrict_init_float() (brevitas.core.restrict_val.FloatRestrictValue method) (brevitas.core.restrict_val.IntRestrictValue method) (brevitas.core.restrict_val.LogFloatRestrictValue method) (brevitas.core.restrict_val.PowerOfTwoRestrictValue method) (brevitas.core.restrict_val.QuantRestrictValue method) restrict_init_inplace_module() (brevitas.core.restrict_val.FloatRestrictValue method) (brevitas.core.restrict_val.IntRestrictValue method) (brevitas.core.restrict_val.LogFloatRestrictValue method) (brevitas.core.restrict_val.PowerOfTwoRestrictValue method) (brevitas.core.restrict_val.QuantRestrictValue method) restrict_init_module() (brevitas.core.restrict_val.FloatRestrictValue method) (brevitas.core.restrict_val.IntRestrictValue method) (brevitas.core.restrict_val.LogFloatRestrictValue method) (brevitas.core.restrict_val.PowerOfTwoRestrictValue method) (brevitas.core.restrict_val.QuantRestrictValue method)

restrict_init_tensor() (brevitas.core.restrict_val.FloatRestrictValue method) (brevitas.core.restrict_val.IntRestrictValue method) (brevitas.core.restrict_val.LogFloatRestrictValue method) (brevitas.core.restrict_val.PowerOfTwoRestrictValue method) (brevitas.core.restrict_val.QuantRestrictValue method) retrocompatibility_op() (brevitas.core.restrict_val.QuantRestrictValue method) round_ste() (in module brevitas.function.ops_ste) round_ste_impl() (in module brevitas.ops.autograd_ste_ops) round_to_zero() (in module brevitas.function.ops) round_to_zero_ste() (in module brevitas.function.ops_ste) round_to_zero_ste_impl() (in module brevitas.ops.autograd_ste_ops) RoundSte (class in brevitas.core.function_wrapper.ops_ste) RoundSteFn (class in brevitas.ops.autograd_ste_ops) RoundToZeroSte (class in brevitas.core.function_wrapper.ops_ste) RoundToZeroSteFn (class in brevitas.ops.autograd_ste_ops) RuntimeDynamicGroupStatsScaling (class in brevitas.core.scaling.runtime) RuntimeStatsScaling (class in brevitas.core.scaling.runtime)

S

saturating_clamp() (brevitas.core.function_wrapper.clamp.FloatClamp method) scalar_clamp_min_ste() (in module brevitas.function.ops_ste) scalar_clamp_min_ste_impl() (in module brevitas.ops.autograd_ste_ops) scalar_clamp_ste() (in module brevitas.function.ops_ste) scalar_clamp_ste_impl() (in module brevitas.ops.autograd_ste_ops) ScalarClamp (class in brevitas.core.function_wrapper.clamp) ScalarClampMinSte (class in brevitas.core.function_wrapper.ops_ste) ScalarClampMinSteFn (class in brevitas.ops.autograd_ste_ops) ScalarClampSteFn (class in brevitas.ops.autograd_ste_ops)

shrink_lp_op() (in module brevitas.core.stats.stats_op) SingleArgStatelessBuffer (class in brevitas.core.utils) SliceTensor (class in brevitas.core.utils) smooth_normalize_distribution() (brevitas.core.stats.stats_op.KLMinimizerThreshold method) state_dict() (brevitas.core.scaling.standalone.ParameterFromRuntimeStatsScaling method) (brevitas.core.scaling.standalone.ParameterFromStatsFromParameterScaling method) (brevitas.core.utils.StatelessBuffer method) StatelessBuffer (class in brevitas.core.utils) StatsFromParameterScaling (class in brevitas.core.scaling.runtime) StatsInputViewShapeImpl (class in brevitas.core.function_wrapper.shape)

T

tensor_clamp() (in module brevitas.function.ops) tensor_clamp_() (in module brevitas.function.ops) tensor_clamp_ste() (in module brevitas.function.ops_ste) tensor_clamp_ste_() (in module brevitas.function.ops_ste) tensor_clamp_ste_impl() (in module brevitas.ops.autograd_ste_ops) TensorClamp (class in brevitas.core.function_wrapper.clamp) TensorClampSte (class in brevitas.core.function_wrapper.ops_ste) TensorClampSteFn (class in brevitas.ops.autograd_ste_ops) ternary_sign_ste() (in module brevitas.function.ops_ste)

ternary_sign_ste_impl() (in module brevitas.ops.autograd_ste_ops) TernaryQuant (class in brevitas.core.quant.ternary) TernarySignSteFn (class in brevitas.ops.autograd_ste_ops) to_int() (brevitas.core.quant.int_base.DecoupledIntQuant method) (brevitas.core.quant.int_base.IntQuant method) training_forward() (brevitas.core.scaling.standalone.ParameterFromRuntimeStatsScaling method) TruncIntQuant (class in brevitas.core.quant.int) TruncMsbScaling (class in brevitas.core.scaling.standalone) TruncScalingWrapper (class in brevitas.core.scaling.standalone)