Contrib ops and Graph Optimization

[tarekziade]

I am implementing a contrib operation that replaces MatMulIntegerToFloat. I could rewrite the graph to have my operation used instead of the other one.

However, the graph optimizer will replace MatMulInteger to using MatMulIntegerToFloat in MatMulIntegerToFloatFusion so my contrib ops will not be used in replacement of MatMulIntegerToFloat since the optimization step occurs when the session is created.

Is there a way around this? Thanks