- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for mse (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow_to_stablehlo/python/integration_test/tensorflow_to_stablehlo_test.py
return x + 1 model = AddOneModel() x_train = tf.constant([1, 2, 3, 4, 5], dtype=tf.float32) y_train = tf.constant([2, 3, 4, 5, 6], dtype=tf.float32) model.compile(optimizer='sgd', loss='mse') model.fit(x_train, y_train, epochs=1) path = tempdir + '/add_one_model' model.save(path) return path class TensorflowToStableHLOTest(test.TestCase): def test_saved_model_to_stablehlo(self):
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 2.7K bytes - Viewed (0)