Searched refs:compute_dtype (Results 1 – 25 of 427) sorted by relevance
12345678910>>...18
271 def compute_dtype(self): member in Policy515 policy.compute_dtype != policy.variable_dtype)518 if (policy is not None and policy.compute_dtype is not None and519 not dtypes.as_dtype(policy.compute_dtype).is_floating):
48 self.assertEqual(policy.compute_dtype, dtype)54 self.assertEqual(policy.compute_dtype, dtype)58 self.assertEqual(policy.compute_dtype, None)
424 self._dtype_policy.compute_dtype != self._dtype_policy.variable_dtype1772 if self._dtype_policy.compute_dtype:1774 self._dtype_policy.compute_dtype)1790 return self._dtype_policy.compute_dtype1804 compute_dtype = self._compute_dtype1805 if (self._autocast and compute_dtype and1806 dtypes.as_dtype(compute_dtype).is_floating):1812 x.dtype.base_dtype.name != compute_dtype):1813 return math_ops.cast(x, compute_dtype)1817 return tensor_spec.TensorSpec(x.shape, compute_dtype, x.name)[all …]
624 self._dtype_policy.compute_dtype != self._dtype_policy.variable_dtype2371 if self._dtype_policy.compute_dtype:2373 self._dtype_policy.compute_dtype)2386 def compute_dtype(self): member in Layer2405 return self._dtype_policy.compute_dtype2410 return self._dtype_policy.compute_dtype2699 if input_list and self._dtype_policy.compute_dtype is None:
197 if self._dtype_policy.compute_dtype != self._dtype_policy.variable_dtype:200 out = math_ops.cast(out, self._dtype_policy.compute_dtype)
6 name: "compute_dtype"
7 name: "compute_dtype"
15 name: "compute_dtype"
14 name: "compute_dtype"