ckernel_linear_checked.h File Reference

#include <stddef.h>

Go to the source code of this file.

Functions
int	linear_rows_checked_f32 (const float *input, size_t input_elements, size_t input_stride, const float *weight, size_t weight_elements, size_t weight_stride, const float *bias, size_t bias_elements, float *output, size_t output_elements, size_t output_stride, size_t rows, size_t input_channels, size_t output_channels)

Function Documentation

linear_rows_checked_f32()

int linear_rows_checked_f32	(	const float *	input,
		size_t	input_elements,
		size_t	input_stride,
		const float *	weight,
		size_t	weight_elements,
		size_t	weight_stride,
		const float *	bias,
		size_t	bias_elements,
		float *	output,
		size_t	output_elements,
		size_t	output_stride,
		size_t	rows,
		size_t	input_channels,
		size_t	output_channels
	)

Definition at line 25 of file linear_rows_checked.c.

{
    if (!input || !weight || !bias || !output ||
        !span_fits(rows, input_stride, input_channels, input_elements) ||
        !span_fits(output_channels, weight_stride, input_channels, weight_elements) ||
        !span_fits(rows, output_stride, output_channels, output_elements) ||
        bias_elements < output_channels)
        return -1;
 
    /* Validate every accessed value and every result before the first write. */
    for (size_t r = 0; r < rows; ++r)
        for (size_t c = 0; c < input_channels; ++c)
            if (!isfinite(input[r * input_stride + c])) return -1;
    for (size_t o = 0; o < output_channels; ++o) {
        if (!isfinite(bias[o])) return -1;
        for (size_t c = 0; c < input_channels; ++c)
            if (!isfinite(weight[o * weight_stride + c])) return -1;
    }
    for (size_t r = 0; r < rows; ++r)
        for (size_t o = 0; o < output_channels; ++o) {
            const double result = dot_bias(input + r * input_stride,
                                           weight + o * weight_stride,
                                           input_channels, bias[o]);
            if (!isfinite(result) || fabs(result) > FLT_MAX) return -1;
        }
    for (size_t r = 0; r < rows; ++r)
        for (size_t o = 0; o < output_channels; ++o)
            output[r * output_stride + o] = (float)dot_bias(
                input + r * input_stride, weight + o * weight_stride,
                input_channels, bias[o]);
    return 0;
}

References dot_bias(), and span_fits().