diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-10 21:30:40 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-10 21:30:40 +0000 |
commit | 133a45c109da5310add55824db21af5239951f93 (patch) | |
tree | ba6ac4c0a950a0dda56451944315d66409923918 /src/lua/lua_tensor.h | |
parent | Initial commit. (diff) | |
download | rspamd-133a45c109da5310add55824db21af5239951f93.tar.xz rspamd-133a45c109da5310add55824db21af5239951f93.zip |
Adding upstream version 3.8.1.upstream/3.8.1upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/lua/lua_tensor.h')
-rw-r--r-- | src/lua/lua_tensor.h | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/src/lua/lua_tensor.h b/src/lua/lua_tensor.h new file mode 100644 index 0000000..2103868 --- /dev/null +++ b/src/lua/lua_tensor.h @@ -0,0 +1,34 @@ +/*- + * Copyright 2020 Vsevolod Stakhov + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +#ifndef RSPAMD_LUA_TENSOR_H +#define RSPAMD_LUA_TENSOR_H + +#define TENSOR_CLASS "rspamd{tensor}" + +typedef float rspamd_tensor_num_t; + +struct rspamd_lua_tensor { + int ndims; + int size; /* overall size (product of dims) */ + int dim[2]; + rspamd_tensor_num_t *data; +}; + +struct rspamd_lua_tensor *lua_check_tensor(lua_State *L, int pos); +struct rspamd_lua_tensor *lua_newtensor(lua_State *L, int ndims, + const int *dim, bool zero_fill, bool own); + +#endif |