Function is_fp_tensor

PATH/core/fp16/utils.py:59–66

(x)

            yield p

def is_fp_tensor(x):
    if is_nested(x):
        # Fast-fail version of all(is_fp_tensor)
        for y in x:
            if not is_fp_tensor(y):
                return False
        return True
    return is_tensor_like(x) and is_floating_point(x)

def is_nested(x):
    return isinstance(x, tuple) or isinstance(x, list)