!12448 fix cropandresize bugs

From: @yeyunpeng2020
Reviewed-by: @hangangqiang,@HilbertDavid
Signed-off-by: @hangangqiang
pull/12448/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit b393206b5f

@ -36,27 +36,20 @@ STATUS TFCropAndResizeParser::Parse(const tensorflow::NodeDef &tf_op,
MS_LOG(ERROR) << "New PrimitiveT failed";
return RET_NULL_PTR;
}
auto attr = std::make_unique<schema::ResizeT>();
auto attr = std::make_unique<schema::CropAndResizeT>();
if (attr == nullptr) {
MS_LOG(ERROR) << "new attr failed";
return RET_NULL_PTR;
}
tensorflow::AttrValue attr_value;
attr->format = schema::Format_NHWC;
attr->coordinateTransformMode = schema::CoordinateTransformMode_TF_CROP_AND_RESIZE;
// align_corners
if (TensorFlowUtils::FindAttrValue(tf_op, "align_corners", &attr_value)) {
attr->alignCorners = true;
}
// extrapolation_value
if (!TensorFlowUtils::FindAttrValue(tf_op, "extrapolation_value", &attr_value)) {
MS_LOG(ERROR) << "The align_corners attr should be specified";
return RET_ERROR;
}
attr->extrapolationValue = attr_value.f();
attr->extrapolation_value = attr_value.f();
// method
if (!TensorFlowUtils::FindAttrValue(tf_op, "method", &attr_value)) {

Loading…
Cancel
Save