Fixes for various MKLDNN unit test failures (#16059)
1. MklLayout pass changes Making workspace type uint8 for MaxPool; Handling duplicate control edge insertion 1) Handles case of inserting duplicate control edge (fixing Mkl layout graph pass unit test) 2) Enables uint8 as workspace tensor type (makes consistent with LRN workspace handling) Workspace tensor type change is also performed in MaxPool and MaxPoolGrad operators. 2. Handling MklReshape failing case MklReshape was failing on a unit test when Mkl layout and Tensorflow layout for input tensors were same, but shape of input tensor and output tensor was different. No reorder is required in such case, but reshape is needed. Before this fix, we were asserting that reorder is performed. 3. Adding support for empty input/filter tensors in Convolution backprop operators
Loading
Please sign in to comment