Use the specificed scope in post quant, test=develop (#25384)

fix_copy_if_different
cc 5 years ago committed by GitHub
parent eb3173e2b6
commit 5c8e79956e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -342,7 +342,8 @@ class PostTrainingQuantization(object):
self._executor.run(program=self._program,
feed=data,
fetch_list=self._fetch_list,
return_numpy=False)
return_numpy=False,
scope=self._scope)
if self._algo == "KL":
self._sample_data(batch_id)
else:

Loading…
Cancel
Save