Module tester

Testing and debugging infrastructure.

Structs

Counts

DifferenceFloat

Error

Match

Mismatch

Enums

Difference

Functions

assert_tensors_match

check_cudnn

Check that the given graph produces the correct outputs as described by check_data, which typically comes from a .bin file next to the .onnx file.

check_tensors_match

eval_cudnn

load_check_data

Load the check data into (batch_size, inputs, expected_outputs).