// RUN: infrtopt %s | FileCheck %s // CHECK-LABEL: @ops func @ops() { %a = pd.feed() {name="input0"} : !infrt.lod_tensor %b = pd.feed() {name="input1"} : !infrt.lod_tensor %d = pd.feed() {name="input3"} : !infrt.lod_tensor<3x4x9xf32, 0> %g = "pd.elementwise_add"(%a, %b) {axis=1:si32} : (!infrt.lod_tensor, !infrt.lod_tensor) -> tensor %h = "pd.abs"(%g):(tensor) -> tensor "pd.fetch"(%h) {name="output"} :(tensor)->() }