Searched refs:override_fn_when_input_bool (Results 1 – 1 of 1) sorted by relevance
462 override_fn_when_input_bool=None, argument502 if dtype == torch.bool and override_fn_when_input_bool is not None:503 return override_fn_when_input_bool(*[load(index) for load in loaders])703 override_fn_when_input_bool=None, argument713 if override_fn_when_input_bool is not None:714 override_fn_when_input_bool = ops_wrapper(override_fn_when_input_bool)719 override_fn_when_input_bool=override_fn_when_input_bool,5755 aten.add, allow_alpha=True, override_fn_when_input_bool="logical_or"5837 aten.bitwise_not, override_fn_when_input_bool="logical_not"5884 sign = register_pointwise(aten.sign, override_fn_when_input_bool="identity")