Searched refs:variable_dtype (Results 1 – 25 of 431) sorted by relevance
12345678910>>...18
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | metrics_utils.py | 309 variable_dtype = list(variables_to_update.values())[0].dtype
311 y_true = math_ops.cast(y_true, dtype=variable_dtype)
312 y_pred = math_ops.cast(y_pred, dtype=variable_dtype)
314 thresholds, dtype=variable_dtype)
349 sample_weight = math_ops.cast(sample_weight, dtype=variable_dtype)
407 math_ops.cast(sample_weight, dtype=variable_dtype), y_pred)
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | policy_test.py | 49 self.assertEqual(policy.variable_dtype, dtype)
55 self.assertEqual(policy.variable_dtype, 'float32')
59 self.assertEqual(policy.variable_dtype, None)
|
| D | policy.py | 255 def variable_dtype(self): member in Policy
515 policy.compute_dtype != policy.variable_dtype)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.mixed_precision.-policy.pbtxt | 14 name: "variable_dtype"
|
| D | tensorflow.keras.mixed_precision.experimental.-policy.pbtxt | 19 name: "variable_dtype"
|
| D | tensorflow.keras.layers.-activation.pbtxt | 119 name: "variable_dtype"
|
| D | tensorflow.keras.layers.-cropping3-d.pbtxt | 119 name: "variable_dtype"
|
| D | tensorflow.keras.layers.-layer.pbtxt | 118 name: "variable_dtype"
|
| D | tensorflow.keras.layers.-gaussian-noise.pbtxt | 119 name: "variable_dtype"
|
| D | tensorflow.keras.layers.-leaky-re-l-u.pbtxt | 119 name: "variable_dtype"
|
| D | tensorflow.keras.layers.-cropping1-d.pbtxt | 119 name: "variable_dtype"
|
| D | tensorflow.keras.layers.-up-sampling1-d.pbtxt | 119 name: "variable_dtype"
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | variable_ops.cc | 50 DataType variable_dtype; in Compile() local
53 ctx->GetVariableTypeAndShape(0, &variable_dtype, &shape)); in Compile()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_utils.py | 117 variable_dtype = None
123 variable_dtype = dtype.base_dtype
135 dtype=variable_dtype,
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | embeddings.py | 197 if self._dtype_policy.compute_dtype != self._dtype_policy.variable_dtype:
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.layers.-cropping2-d.pbtxt | 119 name: "variable_dtype"
|
| D | tensorflow.keras.layers.-masking.pbtxt | 119 name: "variable_dtype"
|
| D | tensorflow.keras.layers.-wrapper.pbtxt | 119 name: "variable_dtype"
|
| D | tensorflow.keras.layers.-gaussian-noise.pbtxt | 119 name: "variable_dtype"
|
| D | tensorflow.keras.layers.-re-l-u.pbtxt | 119 name: "variable_dtype"
|
| D | tensorflow.keras.layers.-p-re-l-u.pbtxt | 119 name: "variable_dtype"
|
| D | tensorflow.keras.layers.-up-sampling2-d.pbtxt | 119 name: "variable_dtype"
|
| D | tensorflow.keras.layers.-permute.pbtxt | 119 name: "variable_dtype"
|
| D | tensorflow.keras.layers.-activity-regularization.pbtxt | 119 name: "variable_dtype"
|
| D | tensorflow.keras.layers.-activation.pbtxt | 119 name: "variable_dtype"
|
12345678910>>...18