Allows to use Criterion as part of computation graph.