Searched refs:should_cast_variables (Results 1 – 5 of 5) sorted by relevance
14 name: "should_cast_variables"
121 def should_cast_variables(self): member in Policy
381 if autocast and self._mixed_precision_policy.should_cast_variables:603 self._mixed_precision_policy.should_cast_variables), (653 input_list, self._mixed_precision_policy.should_cast_variables):
1027 if not self._mixed_precision_policy.should_cast_variables: