* fixed the missing implementation of output tensor construction based on type checking * cleanup * cleanup