| graph(%1 : Double(3, 10) |
| %2 : Double(3, 20) |
| %3 : Double(3, 20) |
| %4 : Double(80, 10) |
| %5 : Double(80, 20) |
| %6 : Double(80) |
| %7 : Double(80)) { |
| %8 : Double(10!, 80!) = Transpose[perm=[1, 0]](%4), uses = [%9.i0]; |
| %9 : UNKNOWN_TYPE = Transpose(%8), uses = [%10.i1]; |
| %10 : Double(3, 80) = FC(%1, %9, %6), uses = [%14.i0]; |
| %11 : Double(20!, 80!) = Transpose[perm=[1, 0]](%5), uses = [%12.i0]; |
| %12 : UNKNOWN_TYPE = Transpose(%11), uses = [%13.i1]; |
| %13 : Double(3, 80) = FC(%2, %12, %7), uses = [%14.i1]; |
| %14 : Double(3, 80) = Add(%10, %13), uses = [%15.i0]; |
| %16 : Double(3!, 20), %17 : Double(3!, 20), %18 : Double(3!, 20), %19 : Double(3!, 20) = Split[split=[20, 20, 20, 20], axis=1](%14), uses = [[%20.i0], [%21.i0], [%22.i0], [%23.i0]]; |
| %20 : Double(3, 20) = Sigmoid(%16), uses = [%25.i0]; |
| %21 : Double(3, 20) = Sigmoid(%17), uses = [%24.i0]; |
| %22 : Double(3, 20) = Tanh(%18), uses = [%25.i1]; |
| %23 : Double(3, 20) = Sigmoid(%19), uses = [%28.i0]; |
| %24 : Double(3, 20) = Mul(%21, %3), uses = [%26.i0]; |
| %25 : Double(3, 20) = Mul(%20, %22), uses = [%26.i1]; |
| %26 : Double(3, 20) = Add(%24, %25), uses = [%27.i0, %0.i1]; |
| %27 : Double(3, 20) = Tanh(%26), uses = [%28.i1]; |
| %28 : Double(3, 20) = Mul(%23, %27), uses = [%0.i0]; |
| return (%28, %26); |
| } |