Update the sample code with API change Reflect the change in https://github.com/google/iree/commit/e67c660ac6573745580ebfcaea6ff97d5390829b Change-Id: I192bb11073604873d4f99f1040ae3006bab8f286