ckernel_strided_unary_checked.h File Reference

#include <stddef.h>

Go to the source code of this file.

Functions
int	leaky_relu_strided_f32_checked (const float *input, size_t input_elements, size_t input_stride, float *output, size_t output_elements, size_t output_stride, size_t rows, size_t columns, float negative_slope)
int	tanh_strided_f32_checked (const float *input, size_t input_elements, size_t input_stride, float *output, size_t output_elements, size_t output_stride, size_t rows, size_t columns)
int	transpose_strided_f32_checked (const float *input, size_t input_elements, size_t input_stride, float *output, size_t output_elements, size_t output_stride, size_t rows, size_t columns)

Function Documentation

leaky_relu_strided_f32_checked()

int leaky_relu_strided_f32_checked	(	const float *	input,
		size_t	input_elements,
		size_t	input_stride,
		float *	output,
		size_t	output_elements,
		size_t	output_stride,
		size_t	rows,
		size_t	columns,
		float	negative_slope
	)

Definition at line 23 of file strided_unary_checked.c.

                                                       {
    size_t ni, no;
    if (!ck_checked_span(rows, columns, input_stride, &ni) ||
        !ck_checked_span(rows, columns, output_stride, &no) ||
        input_elements < ni || output_elements < no)
        return -2;
    if (!isfinite(negative_slope) || negative_slope < 0 ||
        !ck_checked_region(input, ni) || !ck_checked_region(output, no) ||
        ck_checked_overlap(input, ni, output, no)) return -1;
    for (size_t row = 0; row < rows; ++row)
        for (size_t column = 0; column < columns; ++column) {
            float value = input[row * input_stride + column];
            if (!isfinite(value) || (value < 0 && !isfinite(value * negative_slope)))
                return -3;
        }
    for (size_t row = 0; row < rows; ++row)
        for (size_t column = 0; column < columns; ++column) {
            float value = input[row * input_stride + column];
            output[row * output_stride + column] =
                value >= 0 ? value : value * negative_slope;
        }
    return 0;
}

References ck_checked_overlap(), ck_checked_region(), and ck_checked_span().

tanh_strided_f32_checked()

int tanh_strided_f32_checked	(	const float *	input,
		size_t	input_elements,
		size_t	input_stride,
		float *	output,
		size_t	output_elements,
		size_t	output_stride,
		size_t	rows,
		size_t	columns
	)

Definition at line 50 of file strided_unary_checked.c.

                                 {
    size_t ni, no;
    if (!ck_checked_span(rows, columns, input_stride, &ni) ||
        !ck_checked_span(rows, columns, output_stride, &no) ||
        input_elements < ni || output_elements < no)
        return -2;
    if (!ck_checked_region(input, ni) || !ck_checked_region(output, no) ||
        ck_checked_overlap(input, ni, output, no)) return -1;
    for (size_t row = 0; row < rows; ++row)
        if (!ck_checked_finite(input + row * input_stride, columns)) return -3;
    for (size_t row = 0; row < rows; ++row)
        for (size_t column = 0; column < columns; ++column)
            output[row * output_stride + column] =
                tanhf(input[row * input_stride + column]);
    return 0;
}

References ck_checked_finite(), ck_checked_overlap(), ck_checked_region(), and ck_checked_span().

transpose_strided_f32_checked()

int transpose_strided_f32_checked	(	const float *	input,
		size_t	input_elements,
		size_t	input_stride,
		float *	output,
		size_t	output_elements,
		size_t	output_stride,
		size_t	rows,
		size_t	columns
	)

Definition at line 4 of file strided_unary_checked.c.

                                 {
    size_t ni, no;
    if (!ck_checked_span(rows, columns, input_stride, &ni) ||
        !ck_checked_span(columns, rows, output_stride, &no) ||
        input_elements < ni || output_elements < no)
        return -2;
    if (!ck_checked_region(input, ni) || !ck_checked_region(output, no) ||
        ck_checked_overlap(input, ni, output, no)) return -1;
    for (size_t row = 0; row < rows; ++row)
        if (!ck_checked_finite(input + row * input_stride, columns)) return -3;
    for (size_t row = 0; row < rows; ++row)
        for (size_t column = 0; column < columns; ++column)
            output[column * output_stride + row] = input[row * input_stride + column];
    return 0;
}

References ck_checked_finite(), ck_checked_overlap(), ck_checked_region(), and ck_checked_span().