Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

[REFACTOR] Refactor test_quantize.py to use Gluon API #20227

Merged
merged 4 commits into from
Jun 17, 2021
Merged

[REFACTOR] Refactor test_quantize.py to use Gluon API #20227

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
58b7d1f
Refactor test_quantization to gluon API
bgawrych Apr 27, 2021
b5b8917
review
bgawrych Apr 28, 2021
69c764f
Apply review suggestions
bgawrych May 25, 2021
d9681b7
Skip flaky test
bgawrych May 26, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 0 additions & 2 deletions src/operator/quantization/mkldnn/mkldnn_quantized_conv.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -38,8 +38,6 @@ static void MKLDNNQuantizedConvForward(const nnvm::NodeAttrs& attrs,
const std::vector<NDArray> &in_data,
const std::vector<OpReqType> &req,
const std::vector<NDArray> &out_data) {
CHECK_EQ(in_data[0].dtype(), mshadow::kUint8)
<< "mkldnn_quantized_conv op only supports uint8 as input type";
TmpMemMgr::Get()->Init(ctx.requested[conv::kTempSpace]);
NDArray weight = in_data[conv::kWeight];
ConvolutionParam param = nnvm::get<ConvolutionParam>(attrs.parsed);
Expand Down
Loading