|
|
|
@@ -74,8 +74,6 @@ STATUS TflitePadParser::Parse(const std::unique_ptr<tflite::OperatorT> &tflite_o |
|
|
|
MS_LOG(ERROR) << "paddingmode:" << tflite_attr->mode << " don't support"; |
|
|
|
return RET_INVALID_OP_ATTR; |
|
|
|
} |
|
|
|
AddOpInput(op, tensors_id, tensors_format, tensors_id_map, tflite_op->inputs[1], tensors_id->size(), |
|
|
|
tflite_tensors.size(), schema::Format::Format_NHWC); |
|
|
|
} else { |
|
|
|
MS_LOG(ERROR) << "this pad:" << node_name << " hasn't been supported"; |
|
|
|
return RET_NOT_SUPPORT; |
|
|
|
@@ -86,6 +84,10 @@ STATUS TflitePadParser::Parse(const std::unique_ptr<tflite::OperatorT> &tflite_o |
|
|
|
|
|
|
|
AddOpInput(op, tensors_id, tensors_format, tensors_id_map, tflite_op->inputs[0], tensors_id->size(), |
|
|
|
tflite_tensors.size(), schema::Format::Format_NHWC); |
|
|
|
if (std::strcmp(node_name, "MirrorPad") == 0) { |
|
|
|
AddOpInput(op, tensors_id, tensors_format, tensors_id_map, tflite_op->inputs[1], tensors_id->size(), |
|
|
|
tflite_tensors.size(), schema::Format::Format_NHWC); |
|
|
|
} |
|
|
|
AddOpOutput(op, tensors_id, tensors_format, tensors_id_map, tflite_op->outputs[0], tensors_id->size(), |
|
|
|
tflite_tensors.size(), schema::Format::Format_NHWC); |
|
|
|
return RET_OK; |
|
|
|
|