Home
last modified time | relevance | path

Searched refs:floatx (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/python/keras/
Dbackend_config_test.py40 floatx = 'float64'
41 backend_config.set_floatx(floatx)
42 self.assertEqual(backend_config.floatx(), floatx)
44 self.assertEqual(backend_config.floatx(), 'float32')
Dlosses.py1248 math_ops.reduce_variance(math_ops.cast(row_lens, K.floatx())),
1538 y_pred = math_ops.cast(y_pred, dtype=K.floatx())
1539 y_true = math_ops.cast(y_true, dtype=K.floatx())
1540 delta = math_ops.cast(delta, dtype=K.floatx())
1623 label_smoothing, dtype=K.floatx())
1721 label_smoothing, dtype=K.floatx())
Dbackend_config.py68 def floatx(): function
Dactivations_test.py109 positive_values = np.array([[1, 2]], dtype=backend.floatx())
113 negative_values = np.array([[-1, -2]], dtype=backend.floatx())
210 negative_values = np.array([[-1, -2]], dtype=backend.floatx())
Dbackend.py172 floatx = backend_config.floatx variable
222 return math_ops.cast(x, dtype=floatx())
223 return np.asarray(x, dtype=floatx())
1079 dtype = floatx()
1229 dtype = floatx()
1332 dtype = floatx()
1597 dtype = floatx()
1633 dtype = floatx()
1667 dtype = floatx()
1764 dtype = floatx()
[all …]
Dmodels_test.py393 floatx = keras.backend.floatx()
404 keras.backend.set_floatx(floatx)
Dconstraints.py94 return w * math_ops.cast(math_ops.greater_equal(w, 0.), K.floatx())
Dmetrics.py163 self._dtype = K.floatx() if dtype is None else dtypes.as_dtype(dtype).name
3291 return math_ops.cast(math_ops.equal(y_true, y_pred), K.floatx())
3348 K.floatx())
3389 return math_ops.cast(math_ops.equal(y_true, y_pred), K.floatx())
3415 nn.in_top_k(y_pred, math_ops.argmax(y_true, axis=-1), k), K.floatx())
3451 nn.in_top_k(y_pred, math_ops.cast(y_true, 'int32'), k), K.floatx())
Doptimizer_v1.py522 t = math_ops.cast(self.iterations, K.floatx())
616 t = math_ops.cast(self.iterations, K.floatx())
698 t = math_ops.cast(self.iterations, K.floatx())
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dcategory_encoding.py143 output_dtype = K.floatx() if self.output_mode == TFIDF else dtypes.int64
201 result = math_ops.cast(result, K.floatx())
217 dtype=K.floatx(),
Dnormalization_v1.py146 dtype=K.floatx(),
151 dtype=K.floatx(),
172 if inputs.dtype != K.floatx():
173 inputs = math_ops.cast(inputs, K.floatx())
Dindex_lookup.py237 dtype=K.floatx(),
255 output_dtype = self._value_dtype if self.output_mode == INT else K.floatx()
Dtext_vectorization.py377 output_dtype = dtypes.int64 if self._output_mode == INT else K.floatx()
/external/tensorflow/tensorflow/python/keras/preprocessing/
Dimage.py189 dtype = backend.floatx()
230 dtype = backend.floatx()
378 dtype = backend.floatx()
447 dtype = backend.floatx()
791 dtype = backend.floatx()
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dpolicy.py447 return Policy(backend.floatx())
/external/tensorflow/tensorflow/python/keras/layers/
Dembeddings.py114 kwargs['dtype'] = K.floatx()
Dpooling.py792 mask = math_ops.cast(mask, backend.floatx())
/external/tensorflow/tensorflow/python/keras/engine/
Dinput_layer.py148 dtype = backend.floatx()
Dinput_spec.py283 default_dtype = default_dtype or backend.floatx()
Dtraining_utils_v1.py1041 class_sample_weight = math_ops.cast(class_sample_weight, K.floatx())
1044 ops.convert_to_tensor_v2_with_dispatch(sample_weight), K.floatx())
1355 dtype = dtype or K.floatx()
1768 dtype = K.floatx()
Ddata_adapter.py863 return np.array(t, dtype=backend.floatx())
1045 dtype = backend.floatx()
1654 data = data.astype(backend.floatx())
Dbase_layer_v1.py387 dtype = self.dtype or backend.floatx()
1039 loss, dtype=backend.floatx())
1055 loss, dtype=backend.floatx())
Dbase_layer_utils.py166 return array_ops.placeholder(shape=shape, dtype=backend.floatx())
/external/tensorflow/tensorflow/python/keras/applications/
Dimagenet_utils.py190 x = x.astype(backend.floatx(), copy=False)
/external/tensorflow/tensorflow/python/keras/initializers/
Dinitializers_v2.py932 dtype = backend.floatx()

12