// RUN: tf-mlir-translate -hlo-text-to-mlir-hlo %s -o - | FileCheck %s | |
HloModule foo | |
// CHECK-LABEL: func @main(%arg0: tensor<16xf32>) -> tensor<16xf32> { | |
ENTRY %foo (arg0.1: f32[16]) -> f32[16] { | |
%arg0.1 = f32[16] parameter(0) | |
// CHECK-NEXT: %0 = "xla_hlo.log"(%arg0) {name = "log.2"} : (tensor<16xf32>) -> tensor<16xf32> | |
// CHECK-NEXT: return %0 : tensor<16xf32> | |
ROOT %log.2 = f32[16] log(f32[16] %arg0.1) | |
} |