`core.fusions.fused_bias_dropout`#

Module Contents#

Functions#

`_bias_dropout_add_func`
`bias_dropout_add_unfused`
`bias_dropout_add_fused_train`
`bias_dropout_add_fused_inference`
`get_bias_dropout_add`

API#

core.fusions.fused_bias_dropout._bias_dropout_add_func( x_with_bias: Tuple[Tensor, Optional[Tensor]], residual: Tensor, prob: float, training: bool, ) → Tensor#

core.fusions.fused_bias_dropout.bias_dropout_add_unfused(training)#

core.fusions.fused_bias_dropout.bias_dropout_add_fused_train( x_with_bias: Tuple[torch.Tensor, Optional[torch.Tensor]], residual: torch.Tensor, prob: float, ) → torch.Tensor#

core.fusions.fused_bias_dropout.bias_dropout_add_fused_inference( x_with_bias: Tuple[torch.Tensor, Optional[torch.Tensor]], residual: torch.Tensor, prob: float, ) → torch.Tensor#

core.fusions.fused_bias_dropout.get_bias_dropout_add(training, fused)#