Lines Matching defs:bool
441 return_indices: bool = False,
518 return_indices: bool = False,
553 return_indices: bool = False,
634 return_indices: bool = False,
670 ceil_mode: bool = False,
671 return_indices: bool = False,
723 ceil_mode: bool = False,
724 return_indices: bool = False,
760 ceil_mode: bool = False,
761 return_indices: bool = False,
813 ceil_mode: bool = False,
814 return_indices: bool = False,
850 ceil_mode: bool = False,
851 return_indices: bool = False,
903 ceil_mode: bool = False,
904 return_indices: bool = False,
1081 ceil_mode: bool = False,
1119 ceil_mode: bool = False,
1155 ceil_mode: bool = False,
1189 return_indices: bool = False,
1217 return_indices: bool = False,
1244 return_indices: bool = False,
1273 return_indices: bool = False,
1300 return_indices: bool = False,
1330 return_indices: bool = False,
1404 training: bool = True,
1405 inplace: bool = False,
1432 training: bool = False,
1433 inplace: bool = False,
1455 training: bool = True,
1456 inplace: bool = False,
1506 training: bool = True,
1507 inplace: bool = False,
1565 training: bool = True,
1566 inplace: bool = False,
1617 training: bool = False,
1618 inplace: bool = False,
1661 inplace: bool = False,
1693 def relu(input: Tensor, inplace: bool = False) -> Tensor: # noqa: D400,D402
1749 inplace: bool = False,
1780 def relu6(input: Tensor, inplace: bool = False) -> Tensor: # noqa: D400,D402
1796 def elu(input: Tensor, alpha: float = 1.0, inplace: bool = False) -> Tensor:
1820 def selu(input: Tensor, inplace: bool = False) -> Tensor: # noqa: D400,D402
1852 inplace: bool = False,
1885 inplace: bool = False,
1941 training: bool = False,
1942 inplace: bool = False,
2149 hard: bool = False,
2287 def hardsigmoid(input: Tensor, inplace: bool = False) -> Tensor:
2358 def silu(input: Tensor, inplace: bool = False) -> Tensor:
2383 def mish(input: Tensor, inplace: bool = False) -> Tensor:
2403 def hardswish(input: Tensor, inplace: bool = False) -> Tensor:
2443 scale_grad_by_freq: bool = False,
2444 sparse: bool = False,
2560 scale_grad_by_freq: bool = False,
2562 sparse: bool = False,
2564 include_last_offset: bool = False,
2787 training: bool = False,
2842 use_input_stats: bool = True,
3012 zero_infinity: bool = False,
3088 size_average: Optional[bool] = None,
3090 reduce: Optional[bool] = None,
3156 log_input: bool = True,
3157 full: bool = False,
3158 size_average: Optional[bool] = None,
3160 reduce: Optional[bool] = None,
3224 full: bool = False,
3309 size_average: Optional[bool] = None,
3310 reduce: Optional[bool] = None,
3312 log_target: bool = False,
3393 size_average: Optional[bool] = None,
3395 reduce: Optional[bool] = None,
3493 size_average: Optional[bool] = None,
3494 reduce: Optional[bool] = None,
3561 size_average: Optional[bool] = None,
3562 reduce: Optional[bool] = None,
3636 size_average: Optional[bool] = None,
3637 reduce: Optional[bool] = None,
3723 size_average: Optional[bool] = None,
3724 reduce: Optional[bool] = None,
3762 size_average: Optional[bool] = None,
3763 reduce: Optional[bool] = None,
3802 size_average: Optional[bool] = None,
3803 reduce: Optional[bool] = None,
3838 size_average: Optional[bool] = None,
3839 reduce: Optional[bool] = None,
3867 size_average: Optional[bool] = None,
3868 reduce: Optional[bool] = None,
3895 size_average: Optional[bool] = None,
3896 reduce: Optional[bool] = None,
3925 size_average: Optional[bool] = None,
3926 reduce: Optional[bool] = None,
3973 size_average: Optional[bool] = None,
3974 reduce: Optional[bool] = None,
4006 size_average: Optional[bool] = None,
4007 reduce: Optional[bool] = None,
4183 align_corners: Optional[bool] = None,
4194 align_corners: Optional[bool] = None,
4294 align_corners: Optional[bool] = None,
4295 recompute_scale_factor: Optional[bool] = None,
4296 antialias: bool = False,
4307 align_corners: Optional[bool] = None,
4308 recompute_scale_factor: Optional[bool] = None,
4309 antialias: bool = False,
4320 align_corners: Optional[bool] = None,
4321 recompute_scale_factor: Optional[bool] = None,
4322 antialias: bool = False,
4333 align_corners: Optional[bool] = None,
4334 recompute_scale_factor: Optional[bool] = None,
4335 antialias: bool = False,
4345 align_corners: Optional[bool] = None,
4346 recompute_scale_factor: Optional[bool] = None,
4347 antialias: bool = False,
4759 align_corners: Optional[bool] = None,
4916 align_corners: Optional[bool] = None,
5231 swap: bool = False,
5232 size_average: Optional[bool] = None,
5233 reduce: Optional[bool] = None,
5273 swap: bool = False,
5838 check_other: bool = True,
5868 def multi_head_attention_forward(
5878 add_zero_attn: bool,
5882 training: bool = True,
5884 need_weights: bool = True,
5886 use_separate_proj_weight: bool = False,