Converting some op usage in graph_tools_test.py to raw_ops to avoid having a condition on static asserts in TF and make tests more robust.
Converting some op usage in graph_tools_test.py to raw_ops to avoid having a condition on static asserts in TF and make tests more robust.

PiperOrigin-RevId: 268743270

tensorflow/transform