Set Vspace only one time
I don't believe there is currently a use-case for a different VSpace (and it doesn't seem to be controllable through any public method). If it is a usecase we want to support, it should be simple enough to add an overload of TFE_Py_TapeGradient. PiperOrigin-RevId: 211917235
Loading
Please sign in to comment