tensor_map.mlir.in

// RUN: infrtexec -i %s | FileCheck %s

func @load_tensor_map() {
  %path = Infrt.get_string("@CMAKE_BINARY_DIR@/multi_fc_model")
  %map = dt.load_params(%path)
  %size = dt.tensor_map_get_size(%map) -> i32
  Infrt.print.i32 %size

  %a = dt.tensor_map_get_tensor(%map) {name="fc_bias"} -> !infrt.dense_tensor<CPU, FP32, NCHW>

  // CHECK: tensor: shape=shape[2], values=[0, 0]
  dt.print_tensor (%a : !infrt.dense_tensor<CPU, FP32, NCHW>)

  Infrt.return
}