pub fn new_tensor<'a, T>(&'a mut self, shape: &'a [u64]) -> impl Tensor + 'a {
      let location = Location::unknown(&self.melior_context);
      let values: Vec<_> = vec![1.0f32, 2.0f32, 3.0f32, 4.0f32]
          .iter()
          .map(|x| {
              FloatAttribute::new(
                  &self.melior_context,
                  *x as f64,
                  Type::float32(&self.melior_context),
              )
              .into()
          })
          .collect();
      let mlir_type = RankedTensorType::new(shape, Type::float32(&self.melior_context), None);
      let attr =
          DenseElementsAttribute::new(mlir_type.into(), &values).unwrap();
      println!("{:?}", attr);
      let op = tosa::r#const(&self.melior_context, attr.into(), location);
      ConstTensor { op, shape }
 }