Searched refs:AutoCastVariable (Results 1 – 3 of 3) sorted by relevance
50 class AutoCastVariable(variables.Variable, core.Tensor): class496 ops.register_tensor_conversion_function(AutoCastVariable,497 … AutoCastVariable._dense_var_to_tensor) # pylint:disable=protected-access518 return AutoCastVariable(variable)520 class AutoCastDistributedVariable(AutoCastVariable, variable.__class__):
159 self.assertIsInstance(var, autocast_variable.AutoCastVariable)
416 * `AutoCastVariable.dtype` now refers to the actual variable dtype, not the431 * In `Layer.call`, `AutoCastVariable`s will no longer be casted within433 variable is used to determine whether `AutoCastVariable`s are casted, and those435 …in `Layer.call`; if one of those two functions calls `Layer.call`, `AutoCastVariable`s will still …