Report the working memory requirements, for both inference and training. As noted in MemoryReport
Working memory is memory That will be allocated in a ND4J workspace, or can be garbage collected at any
points after the method returns.

Parameters:

fixedInference - Number of elements of working memory used for inference (independent of minibatch size)

variableInferencePerEx - Number of elements of working memory used for inference, for each example

fixedTrain - Number of elements of working memory used for training (independent of
minibatch size), for each cache mode

variableTrainPerEx - Number of elements of working memory used for training, for each example, for
each cache mode