|
|
@ -797,7 +797,7 @@ void update_shape_flatten(const std::vector<DLTensor *> &tensors, int64_t *shape
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
std::string GetEltwiseMode(const OpCommonAttr &opAttr, const mindspore::predict::OpDef &opdef) {
|
|
|
|
std::string GetEltwiseMode(const OpCommonAttr &opAttr, const mindspore::predict::OpDef &opdef) {
|
|
|
|
auto &optype = opAttr.optype;
|
|
|
|
const auto optype = opAttr.optype;
|
|
|
|
std::string mode = "add";
|
|
|
|
std::string mode = "add";
|
|
|
|
if (optype == "Eltwise") {
|
|
|
|
if (optype == "Eltwise") {
|
|
|
|
auto op_mode = opdef.attr_as_Eltwise()->mode();
|
|
|
|
auto op_mode = opdef.attr_as_Eltwise()->mode();
|
|
|
|