GRAPH API MOTensor not working

from max import graph

var m = graph.Module()

var g = m.graph(
    "my_model",
    in_types=MOTensor(DType.float32, 1, 28, 28, 1),
    out_types=MOTensor(DType.float32, 1, 10),
)

 error: use of unknown declaration 'MOTensor'
    in_types= MOTensor(DType.float32, 1, 28, 28, 1)