General utilities for code generation.
|
|
|
row_major_index_string(i,
shape)
Constructs an index string for a row major (C type) indexing of a
flattened tensor of rank 0, 1, or 2. |
source code
|
|
|
optimize_floats(code)
Optimize storage size of floating point numbers by removing unneeded
trailing zeros. |
source code
|
|
|
function_signature(name,
return_type="void",
args=[],
virtual=False,
inline=False,
const=False,
classname=None)
Render function signature from arguments. |
source code
|
|
|
gen_token_code(token,
rule=default_code_rule) |
source code
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
gen_switch(argument,
cases,
default_case=None,
braces=False) |
source code
|
|
|
|
|
|
|
|