Tensor Comprehensions
Main Page
Namespaces
Classes
Files
File List
File Members
All
Classes
Namespaces
Files
Functions
Variables
Typedefs
Enumerations
Enumerator
Friends
Macros
Pages
dper_lut_concat.h
Go to the documentation of this file.
1
16
#pragma once
17
18
namespace
tc {
19
20
constexpr
static
auto
TC_DPER_LUT_CONCAT_NAME =
"dper_lut_concat"
;
21
22
constexpr
static
auto
TC_DPER_LUT_CONCAT = R
"TC( def dper_lut_concat(float(B, M) I1, int32(B, L1) I2, int32(B, L2) I3, float(N, M) W1, float(N) B1, float(E1, D) LUT1, float(E2, D) LUT2) -> (O1, O2, O3) { O1(b, n) +=! I1(b, m) * W1(n, m) O1(b, n) = O1(b, n) + B1(n) O2(i, j) +=! LUT1(I2(i, k), j) O3(i, j) +=! LUT2(I3(i, k), j) } )TC";
23
24
}
// namespace tc
25
include
tc
library
dper_lut_concat.h
Generated by
1.8.5