Q.ANT Native Computing Toolkit 2.3.1
All you need for photonic computing
Loading...
Searching...
No Matches
qant_native_computing_toolkit_native.h
Go to the documentation of this file.
1#ifndef QANT_NATIVE_COMPUTING_TOOLKIT_NATIVE_H
2#define QANT_NATIVE_COMPUTING_TOOLKIT_NATIVE_H
3
4#include <stdint.h>
5
6#include <dlpack/dlpack.h>
7
8#ifdef __cplusplus
9namespace qant_native_computing_toolkit::native
10{
11 extern "C"
12 {
13#endif
14
26 DLManagedTensorVersioned *mul_npu(const uint32_t npu_id, DLManagedTensorVersioned const *us,
27 DLManagedTensorVersioned const *vs);
28
40 DLManagedTensorVersioned *calc_scaled_periodic_nl_fprop(const uint32_t npu_id, DLManagedTensorVersioned const *features, DLManagedTensorVersioned const *weights);
41
62 DLManagedTensorVersioned *linear_fprop(const uint32_t npu_id, DLManagedTensorVersioned const *features,
63 DLManagedTensorVersioned const *weights);
64
78 DLManagedTensorVersioned *mul_npu_f32(const uint32_t npu_id, DLManagedTensorVersioned const *us,
79 DLManagedTensorVersioned const *vs);
80
94 DLManagedTensorVersioned *mul_npu_i16(const uint32_t npu_id, DLManagedTensorVersioned const *us,
95 DLManagedTensorVersioned const *vs);
96
97#ifdef __cplusplus
98 } // extern "C"
99} // namespace qant_native_computing_toolkit::native
100#endif
101
102#endif // QANT_NATIVE_COMPUTING_TOOLKIT_NATIVE_H
linear_fprop
DLManagedTensorVersioned * linear_fprop(const uint32_t npu_id, DLManagedTensorVersioned const *features, DLManagedTensorVersioned const *weights)
Performs a linear forward propagation between input features and a weight matrix.
mul_npu
DLManagedTensorVersioned * mul_npu(const uint32_t npu_id, DLManagedTensorVersioned const *us, DLManagedTensorVersioned const *vs)
Multiplies two 1D Tensors element-wise.
calc_scaled_periodic_nl_fprop
DLManagedTensorVersioned * calc_scaled_periodic_nl_fprop(const uint32_t npu_id, DLManagedTensorVersioned const *features, DLManagedTensorVersioned const *weights)
Calculates a scaled periodic nonlinearity pairwise for all elements of features and weights .
mul_npu_i16
DLManagedTensorVersioned * mul_npu_i16(const uint32_t npu_id, DLManagedTensorVersioned const *us, DLManagedTensorVersioned const *vs)
Multiplies two 1D Tensors of int16 element-wise.
mul_npu_f32
DLManagedTensorVersioned * mul_npu_f32(const uint32_t npu_id, DLManagedTensorVersioned const *us, DLManagedTensorVersioned const *vs)
Multiplies two 1D Tensors of float32 element-wise.