def register_arithmetic_ufunc(register_inferer, register_unbound, binary_ufunc): register_inferer(np, binary_ufunc, binary_map) register_unbound(np, binary_ufunc, "reduce", reduce_) register_unbound(np, binary_ufunc, "accumulate", accumulate) register_unbound(np, binary_ufunc, "reduceat", reduceat) register_unbound(np, binary_ufunc, "outer", outer)
'bitwise_or', 'bitwise_xor', 'left_shift', 'right_shift', ) binary_ufuncs_arithmetic = ( # Arithmetic ufuncs 'add', 'subtract', 'multiply', 'true_divide', 'floor_divide', 'divide', ) #------------------------------------------------------------------------ # Register our type functions #------------------------------------------------------------------------ register_inferer(np, 'sum', reduce_) register_inferer(np, 'prod', reduce_) for binary_ufunc in binary_ufuncs_bitwise + binary_ufuncs_arithmetic: register_inferer(np, binary_ufunc, binary_map) register_unbound(np, binary_ufunc, "reduce", reduce_) for binary_ufunc in binary_ufuncs_compare + binary_ufuncs_logical: register_inferer(np, binary_ufunc, binary_map_bool) register_unbound(np, binary_ufunc, "reduce", reduce_bool)
def register_bool_ufunc(register_inferer, register_unbound, binary_ufunc): register_inferer(np, binary_ufunc, binary_map_bool) register_unbound(np, binary_ufunc, "reduce", reduce_bool) register_unbound(np, binary_ufunc, "accumulate", accumulate_bool) register_unbound(np, binary_ufunc, "reduceat", reduceat_bool) register_unbound(np, binary_ufunc, "outer", outer_bool)
'add', 'subtract', 'multiply', 'true_divide', 'floor_divide', ) if not PY3: binary_ufuncs_arithmetic = binary_ufuncs_arithmetic + ('divide', ) #------------------------------------------------------------------------ # Register our type functions #------------------------------------------------------------------------ register_inferer(np, 'sum', reduce_) register_inferer(np, 'prod', reduce_) for binary_ufunc in binary_ufuncs_bitwise + binary_ufuncs_arithmetic: register_inferer(np, binary_ufunc, binary_map) register_unbound(np, binary_ufunc, "reduce", reduce_) register_unbound(np, binary_ufunc, "accumulate", accumulate) register_unbound(np, binary_ufunc, "reduceat", reduceat) register_unbound(np, binary_ufunc, "outer", outer) for binary_ufunc in binary_ufuncs_compare + binary_ufuncs_logical: register_inferer(np, binary_ufunc, binary_map_bool) register_unbound(np, binary_ufunc, "reduce", reduce_bool) register_unbound(np, binary_ufunc, "accumulate", accumulate_bool) register_unbound(np, binary_ufunc, "reduceat", reduceat_bool) register_unbound(np, binary_ufunc, "outer", outer_bool)
def register_bool_ufunc(register_inferer, register_unbound, binary_ufunc): register_inferer(np, binary_ufunc, binary_map_bool) register_unbound(np, binary_ufunc, "reduce", reduce_bool) register_unbound(np, binary_ufunc, "accumulate", accumulate_bool) register_unbound(np, binary_ufunc, "reduceat", reduceat_bool) register_unbound(np, binary_ufunc, "outer", outer_bool)
def register_arithmetic_ufunc(register_inferer, register_unbound, binary_ufunc): register_inferer(np, binary_ufunc, binary_map) register_unbound(np, binary_ufunc, "reduce", reduce_) register_unbound(np, binary_ufunc, "accumulate", accumulate) register_unbound(np, binary_ufunc, "reduceat", reduceat) register_unbound(np, binary_ufunc, "outer", outer)