Commit 6d36f1b4 authored by Gaurav Jain's avatar Gaurav Jain Committed by TensorFlower Gardener
Browse files

Make tests compatible with TF2 

Handle TensorShapeV2 changes as well as assertion behavior changes.

PiperOrigin-RevId: 235935411
parent 0c8deb2f
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment