fix import error in cpu environment

Commit

1 year ago

fix import error in cpu environment Signed-off-by: Masahiro Tanaka <mtanaka@microsoft.com>