-
Notifications
You must be signed in to change notification settings - Fork 18
/
ops.h
17 lines (12 loc) · 816 Bytes
/
ops.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#pragma once
#include "ggml.h"
struct ggml_tensor *pad_1d(struct ggml_context *ctx0, struct ggml_tensor *inp,
int padding_left, int padding_right);
struct ggml_tensor *unpad_1d(struct ggml_context *ctx0, struct ggml_tensor *inp,
int padding_left, int padding_right);
struct ggml_tensor *strided_conv_1d(struct ggml_context *ctx0, struct ggml_tensor *inp,
struct ggml_tensor *conv_w, struct ggml_tensor *conv_b,
int stride);
struct ggml_tensor *strided_conv_transpose_1d(struct ggml_context *ctx0, struct ggml_tensor *inp,
struct ggml_tensor *conv_w, struct ggml_tensor *conv_b,
int stride);