- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for test_tf_mlir_to_stablehlo (0.15 sec)
-
tensorflow/compiler/mlir/tensorflow_to_stablehlo/python/integration_test/tensorflow_to_stablehlo_test.py
with ir.Context() as ctx: stablehlo.register_dialect(ctx) module = ir.Module.parse(module_bytecode) self.assertIn('stablehlo.add %arg0, %cst : tensor<4xf32>', str(module)) def test_tf_mlir_to_stablehlo(self): assembly = """ module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 268 : i32}} { func.func @main(%arg0 : tensor<?xf32>) -> tensor<?xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 2.7K bytes - Viewed (0)