Browse Source

fix bug of floor_div

tags/v1.0.0
tao_yunhao 5 years ago
parent
commit
aac49405a1
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/src/ops/floor_div.cc

+ 1
- 1
mindspore/lite/src/ops/floor_div.cc View File

@@ -24,7 +24,7 @@ int FloorDiv::UnPackToFlatBuilder(const schema::Primitive *primitive, flatbuffer
MS_ASSERT(nullptr != primitive);
MS_ASSERT(nullptr != fbb);
auto val_offset = schema::CreateFloor(*fbb);
auto prim_offset = schema::CreatePrimitive(*fbb, schema::PrimitiveType_Floor, val_offset.o);
auto prim_offset = schema::CreatePrimitive(*fbb, schema::PrimitiveType_FloorDiv, val_offset.o);
fbb->Finish(prim_offset);
return RET_OK;
}


Loading…
Cancel
Save