fix generate_mask_labels lod level (#22743)

revert-22710-feature/integrated_ps_api
dyning 5 years ago committed by GitHub
parent ba140222d6
commit 1c0653462d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -75,6 +75,11 @@ class GenerateMaskLabelsOp : public framework::OperatorWithKernel {
ctx->SetOutputDim("MaskRois", {-1, 4}); ctx->SetOutputDim("MaskRois", {-1, 4});
ctx->SetOutputDim("RoiHasMaskInt32", {-1, 1}); ctx->SetOutputDim("RoiHasMaskInt32", {-1, 1});
ctx->SetOutputDim("MaskInt32", {-1, num_classes * resolution * resolution}); ctx->SetOutputDim("MaskInt32", {-1, num_classes * resolution * resolution});
if (!ctx->IsRuntime()) {
ctx->SetLoDLevel("MaskRois", ctx->GetLoDLevel("Rois"));
ctx->SetLoDLevel("RoiHasMaskInt32", ctx->GetLoDLevel("Rois"));
ctx->SetLoDLevel("MaskInt32", ctx->GetLoDLevel("Rois"));
}
} }
protected: protected:

@ -21,7 +21,6 @@
COMPILE_RUN_OP_WHITE_LIST = [ COMPILE_RUN_OP_WHITE_LIST = [
'sequence_pool', \ 'sequence_pool', \
'sequence_slice', \ 'sequence_slice', \
'generate_mask_labels', \
'generate_proposals', \ 'generate_proposals', \
'mine_hard_examples', \ 'mine_hard_examples', \
'retinanet_detection_output', \ 'retinanet_detection_output', \

Loading…
Cancel
Save