Details
-
New Feature
-
Status: In Progress
-
Major
-
Resolution: Unresolved
-
None
-
None
-
None
-
None
Description
We should add a comparison against TensorFlow in nn/test/test.dml, like we have for other layers, and check that the same values are produced on CPU & GPU for both TensorFlow and SystemML. There is already a conv2d function in nn/test/test.dml that compares the output of three different 2D convolution layers that we have (built-in, DML im2col, DML basic). We should add a new section to this conv2d function (similar to how the max_pool2d test function has multiple test sections) to compare the built-in conv2d layer against the output from TensorFlow. We should use the same X, W, and b initialization values as used in conv2d_transpose.