func @fc( // TODO: Implement "reshape" so that %image can be passed as batch of 2D tensors. %image: tensor, %weights: tensor, %biases: tensor) -> ( tensor ) { %0 = tcf.matmul %weights, %image : (tensor, tensor) -> tensor %1 = tcf.add %0, %biases : (tensor, tensor) -> tensor // TODO: Implement softmax for classification. // For now, this returns a not-terribly useful number. return %1 : tensor }