← Back to C-Kernel-Engine Docs Doxygen Source Documentation
 
Loading...
Searching...
No Matches
Functions
audio_prosody_upsample_checked.c File Reference
#include "ckernel_audio_prosody_upsample_checked.h"
#include "checked_float_buffer.h"
#include <stdint.h>
#include <string.h>

Go to the source code of this file.

Functions

int audio_conv_transpose1d_depthwise_channel_major_f32_checked (const float *input, size_t input_elements, size_t input_stride, const float *weight, size_t weight_elements, const float *bias, size_t bias_elements, float *output, size_t output_elements, size_t output_stride, float *scratch, size_t scratch_elements, size_t channels, size_t input_frames, size_t kernel_size, size_t stride, size_t padding, size_t output_padding, size_t output_frames)
 
int audio_upsample_nearest_channel_major_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 channels, size_t input_frames, size_t factor, size_t output_frames)
 

Function Documentation

◆ audio_conv_transpose1d_depthwise_channel_major_f32_checked()

int audio_conv_transpose1d_depthwise_channel_major_f32_checked ( const float *  input,
size_t  input_elements,
size_t  input_stride,
const float *  weight,
size_t  weight_elements,
const float *  bias,
size_t  bias_elements,
float *  output,
size_t  output_elements,
size_t  output_stride,
float *  scratch,
size_t  scratch_elements,
size_t  channels,
size_t  input_frames,
size_t  kernel_size,
size_t  stride,
size_t  padding,
size_t  output_padding,
size_t  output_frames 
)

Definition at line 37 of file audio_prosody_upsample_checked.c.

45 {
46 size_t input_span, output_span, weight_need, scratch_need, geometry;
47 if (!channels || !input_frames || !kernel_size || !stride ||
48 output_padding >= stride || !output_frames)
49 return CK_AUDIO_UPSAMPLE_INVALID;
50 if (input_frames - 1 > SIZE_MAX / stride)
51 return CK_AUDIO_UPSAMPLE_OVERFLOW;
52 geometry = (input_frames - 1) * stride;
53 if (geometry > SIZE_MAX - kernel_size ||
54 geometry + kernel_size > SIZE_MAX - output_padding ||
55 padding > SIZE_MAX / 2)
56 return CK_AUDIO_UPSAMPLE_OVERFLOW;
57 geometry += kernel_size + output_padding;
58 if (2 * padding >= geometry || geometry - 2 * padding != output_frames)
59 return CK_AUDIO_UPSAMPLE_INVALID;
60 if (!ck_checked_span(channels, input_frames, input_stride, &input_span) ||
61 !ck_checked_span(channels, output_frames, output_stride, &output_span) ||
62 channels > SIZE_MAX / kernel_size ||
63 channels > SIZE_MAX / output_frames)
64 return CK_AUDIO_UPSAMPLE_OVERFLOW;
65 weight_need = channels * kernel_size;
66 scratch_need = channels * output_frames;
67 if (weight_need > SIZE_MAX / sizeof(float) ||
68 scratch_need > SIZE_MAX / sizeof(float))
69 return CK_AUDIO_UPSAMPLE_OVERFLOW;
70 if (input_elements < input_span || weight_elements < weight_need ||
71 bias_elements < channels || output_elements < output_span ||
72 scratch_elements < scratch_need)
73 return CK_AUDIO_UPSAMPLE_CAPACITY;
74 if (!ck_checked_region(input, input_span) ||
75 !ck_checked_region(weight, weight_need) ||
76 !ck_checked_region(bias, channels) ||
77 !ck_checked_region(output, output_span) ||
78 !ck_checked_region(scratch, scratch_need) ||
79 ck_checked_overlap(output, output_span, input, input_span) ||
80 ck_checked_overlap(output, output_span, weight, weight_need) ||
81 ck_checked_overlap(output, output_span, bias, channels) ||
82 ck_checked_overlap(output, output_span, scratch, scratch_need) ||
83 ck_checked_overlap(scratch, scratch_need, input, input_span) ||
84 ck_checked_overlap(scratch, scratch_need, weight, weight_need) ||
85 ck_checked_overlap(scratch, scratch_need, bias, channels))
86 return CK_AUDIO_UPSAMPLE_INVALID;
87 if (!ck_checked_finite(weight, weight_need) ||
88 !ck_checked_finite(bias, channels))
89 return CK_AUDIO_UPSAMPLE_NONFINITE;
90 for (size_t channel = 0; channel < channels; ++channel)
91 if (!ck_checked_finite(input + channel * input_stride, input_frames))
92 return CK_AUDIO_UPSAMPLE_NONFINITE;
93 for (size_t channel = 0; channel < channels; ++channel) {
94 float *row = scratch + channel * output_frames;
95 for (size_t frame = 0; frame < output_frames; ++frame)
96 row[frame] = bias[channel];
97 for (size_t frame = 0; frame < input_frames; ++frame)
98 for (size_t tap = 0; tap < kernel_size; ++tap) {
99 size_t target = frame * stride + tap;
100 if (target >= padding && target - padding < output_frames)
101 row[target - padding] = fmaf(
102 input[channel * input_stride + frame],
103 weight[channel * kernel_size + tap],
104 row[target - padding]);
105 }
106 if (!ck_checked_finite(row, output_frames))
107 return CK_AUDIO_UPSAMPLE_NONFINITE;
108 }
109 for (size_t channel = 0; channel < channels; ++channel)
110 memcpy(output + channel * output_stride,
111 scratch + channel * output_frames,
112 output_frames * sizeof(float));
113 return CK_AUDIO_UPSAMPLE_OK;
114}
ck_checked_finite
static int ck_checked_finite(const float *ptr, size_t n)
Definition checked_float_buffer.h:19
ck_checked_span
static int ck_checked_span(size_t rows, size_t width, size_t stride, size_t *span)
Definition checked_float_buffer.h:6
ck_checked_region
static int ck_checked_region(const float *ptr, size_t n)
Definition checked_float_buffer.h:11
ck_checked_overlap
static int ck_checked_overlap(const float *a, size_t na, const float *b, size_t nb)
Definition checked_float_buffer.h:15
CK_AUDIO_UPSAMPLE_OVERFLOW
@ CK_AUDIO_UPSAMPLE_OVERFLOW
Definition ckernel_audio_prosody_upsample_checked.h:15
CK_AUDIO_UPSAMPLE_CAPACITY
@ CK_AUDIO_UPSAMPLE_CAPACITY
Definition ckernel_audio_prosody_upsample_checked.h:13
CK_AUDIO_UPSAMPLE_INVALID
@ CK_AUDIO_UPSAMPLE_INVALID
Definition ckernel_audio_prosody_upsample_checked.h:12
CK_AUDIO_UPSAMPLE_NONFINITE
@ CK_AUDIO_UPSAMPLE_NONFINITE
Definition ckernel_audio_prosody_upsample_checked.h:14
CK_AUDIO_UPSAMPLE_OK
@ CK_AUDIO_UPSAMPLE_OK
Definition ckernel_audio_prosody_upsample_checked.h:11

References CK_AUDIO_UPSAMPLE_CAPACITY, CK_AUDIO_UPSAMPLE_INVALID, CK_AUDIO_UPSAMPLE_NONFINITE, CK_AUDIO_UPSAMPLE_OK, CK_AUDIO_UPSAMPLE_OVERFLOW, ck_checked_finite(), ck_checked_overlap(), ck_checked_region(), and ck_checked_span().

◆ audio_upsample_nearest_channel_major_f32_checked()

int audio_upsample_nearest_channel_major_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  channels,
size_t  input_frames,
size_t  factor,
size_t  output_frames 
)

Definition at line 7 of file audio_prosody_upsample_checked.c.

11 {
12 size_t input_span, output_span;
13 if (!channels || !input_frames || !factor)
14 return CK_AUDIO_UPSAMPLE_INVALID;
15 if (input_frames > SIZE_MAX / factor)
16 return CK_AUDIO_UPSAMPLE_OVERFLOW;
17 if (input_frames * factor != output_frames ||
18 !ck_checked_span(channels, input_frames, input_stride, &input_span) ||
19 !ck_checked_span(channels, output_frames, output_stride, &output_span))
20 return CK_AUDIO_UPSAMPLE_INVALID;
21 if (input_elements < input_span || output_elements < output_span)
22 return CK_AUDIO_UPSAMPLE_CAPACITY;
23 if (!ck_checked_region(input, input_span) ||
24 !ck_checked_region(output, output_span) ||
25 ck_checked_overlap(input, input_span, output, output_span))
26 return CK_AUDIO_UPSAMPLE_INVALID;
27 for (size_t channel = 0; channel < channels; ++channel)
28 if (!ck_checked_finite(input + channel * input_stride, input_frames))
29 return CK_AUDIO_UPSAMPLE_NONFINITE;
30 for (size_t channel = 0; channel < channels; ++channel)
31 for (size_t frame = 0; frame < output_frames; ++frame)
32 output[channel * output_stride + frame] =
33 input[channel * input_stride + frame / factor];
34 return CK_AUDIO_UPSAMPLE_OK;
35}

References CK_AUDIO_UPSAMPLE_CAPACITY, CK_AUDIO_UPSAMPLE_INVALID, CK_AUDIO_UPSAMPLE_NONFINITE, CK_AUDIO_UPSAMPLE_OK, CK_AUDIO_UPSAMPLE_OVERFLOW, ck_checked_finite(), ck_checked_overlap(), ck_checked_region(), and ck_checked_span().