MLCompilerBridge
Tools for streamlining communication with ML models for compiler optimizations.
|
Classes | |
class | TensorSpec |
class | TensorValue |
Functions | |
TensorValue | read_tensor (io.BufferedReader fs, TensorSpec ts) |
pretty_print_tensor_value (TensorValue tv) | |
read_header (io.BufferedReader f) | |
read_one_observation (Optional[str] context, str event_str, io.BufferedReader f, List[TensorSpec] tensor_specs, Optional[TensorSpec] score_spec) | |
read_stream (str fname) | |
read_stream2 (io.BufferedReader f) | |
main (args) | |
Variables | |
dict | _element_types |
compilerinterface.log_reader.main | ( | args | ) |
Definition at line 144 of file log_reader.py.
compilerinterface.log_reader.pretty_print_tensor_value | ( | TensorValue | tv | ) |
Definition at line 89 of file log_reader.py.
compilerinterface.log_reader.read_header | ( | io.BufferedReader | f | ) |
Definition at line 93 of file log_reader.py.
compilerinterface.log_reader.read_one_observation | ( | Optional[str] | context, |
str | event_str, | ||
io.BufferedReader | f, | ||
List[TensorSpec] | tensor_specs, | ||
Optional[TensorSpec] | score_spec ) |
Definition at line 102 of file log_reader.py.
compilerinterface.log_reader.read_stream | ( | str | fname | ) |
Definition at line 116 of file log_reader.py.
compilerinterface.log_reader.read_stream2 | ( | io.BufferedReader | f | ) |
Definition at line 130 of file log_reader.py.
TensorValue compilerinterface.log_reader.read_tensor | ( | io.BufferedReader | fs, |
TensorSpec | ts ) |
Definition at line 82 of file log_reader.py.
|
protected |
Definition at line 23 of file log_reader.py.