Visualize ttir graph in tt-explorer

[tapspatel]

The following ttir graph gives the error in tt-explorer

AttributeError: 'ttmlir._mlir_libs._mlir.ir.ArrayAttr' object has no attribute 'value'

module {
  func.func @main(%arg0: tensor<1x784xf32>, %arg1: tensor<784x128xf32>, %arg2: tensor<128xf32>, %arg3: tensor<1xi32>) -> tensor<1x128xf32> {
    %0 = tensor.empty() : tensor<1x128xf32>
    %1 = "ttir.matmul"(%arg0, %arg1, %0) : (tensor<1x784xf32>, tensor<784x128xf32>, tensor<1x128xf32>) -> tensor<1x128xf32>
    %2 = tensor.empty() : tensor<1xf32>
    %3 = "ttir.typecast"(%arg3, %2) <{operandSegmentSizes = array<i32: 1, 1>}> : (tensor<1xi32>, tensor<1xf32>) -> tensor<1xf32>
    %4 = tensor.empty() : tensor<1x128xf32>
    %5 = "ttir.broadcast"(%1, %4) <{dimension = [0, 1]}> : (tensor<1x128xf32>, tensor<1x128xf32>) -> tensor<1x128xf32>
    %6 = tensor.empty() : tensor<1x128xf32>
    %7 = "ttir.broadcast"(%3, %6) <{dimension = []}> : (tensor<1xf32>, tensor<1x128xf32>) -> tensor<1x128xf32>
    %8 = tensor.empty() : tensor<1x128xf32>
    %9 = "ttir.multiply"(%5, %7, %8) <{operandSegmentSizes = array<i32: 2, 1>}> : (tensor<1x128xf32>, tensor<1x128xf32>, tensor<1x128xf32>) -> tensor<1x128xf32>
    %10 = tensor.empty() : tensor<1x128xf32>
    %11 = "ttir.broadcast"(%9, %10) <{dimension = [0, 1]}> : (tensor<1x128xf32>, tensor<1x128xf32>) -> tensor<1x128xf32>
    %12 = tensor.empty() : tensor<1x128xf32>
    %13 = "ttir.broadcast"(%arg2, %12) <{dimension = [1]}> : (tensor<128xf32>, tensor<1x128xf32>) -> tensor<1x128xf32>
    %14 = tensor.empty() : tensor<1x128xf32>
    %15 = "ttir.add"(%11, %13, %14) <{operandSegmentSizes = array<i32: 2, 1>}> : (tensor<1x128xf32>, tensor<1x128xf32>, tensor<1x128xf32>) -> tensor<1x128xf32>
    return %15 : tensor<1x128xf32>
  }
}

[odjuricicTT]

@tapspatel Thanks for filing this! Sorry for the late reply i've been OOO. In the future feel free to add @vprajapati-tt to new issues as well.

[odjuricicTT]

@vprajapati-tt Can you take a look at what is causing this and add a test for this case?