|
@@ -54,31 +54,13 @@ class UtestModelManagerModelManager : public testing::Test { |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
void SetUp() {} |
|
|
void SetUp() {} |
|
|
|
|
|
|
|
|
void TearDown() {} |
|
|
void TearDown() {} |
|
|
|
|
|
|
|
|
void CreateGraph(Graph &graph) { |
|
|
|
|
|
TensorDesc desc(ge::Shape({1, 3, 224, 224})); |
|
|
|
|
|
uint32_t size = desc.GetShape().GetShapeSize(); |
|
|
|
|
|
desc.SetSize(size); |
|
|
|
|
|
auto data = op::Data("Data").set_attr_index(0); |
|
|
|
|
|
data.update_input_desc_data(desc); |
|
|
|
|
|
data.update_output_desc_out(desc); |
|
|
|
|
|
|
|
|
|
|
|
auto flatten = op::Flatten("Flatten").set_input_x(data, data.name_out_out()); |
|
|
|
|
|
|
|
|
|
|
|
std::vector<Operator> inputs{data}; |
|
|
|
|
|
std::vector<Operator> outputs{flatten}; |
|
|
|
|
|
std::vector<Operator> targets{flatten}; |
|
|
|
|
|
// Graph graph("test_graph"); |
|
|
|
|
|
graph.SetInputs(inputs).SetOutputs(outputs).SetTargets(targets); |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
void GenUnencryptModelData(ModelData &data) { |
|
|
void GenUnencryptModelData(ModelData &data) { |
|
|
const int model_len = 10; |
|
|
const int model_len = 10; |
|
|
data.model_len = sizeof(ModelFileHeader) + model_len; |
|
|
data.model_len = sizeof(ModelFileHeader) + model_len; |
|
|
data.model_data = new uint8_t[data.model_len]; |
|
|
data.model_data = new uint8_t[data.model_len]; |
|
|
memset((uint8_t *)data.model_data + sizeof(ModelFileHeader), 0, model_len); |
|
|
|
|
|
|
|
|
memset(data.model_data, 0, data.model_len); |
|
|
|
|
|
|
|
|
ModelFileHeader *header = (ModelFileHeader *)data.model_data; |
|
|
ModelFileHeader *header = (ModelFileHeader *)data.model_data; |
|
|
header->magic = MODEL_FILE_MAGIC_NUM; |
|
|
header->magic = MODEL_FILE_MAGIC_NUM; |
|
@@ -88,19 +70,6 @@ class UtestModelManagerModelManager : public testing::Test { |
|
|
header->is_checksum = ModelCheckType::CHECK; |
|
|
header->is_checksum = ModelCheckType::CHECK; |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
void GenEncryptModelData(ModelData &data) { |
|
|
|
|
|
const int model_len = 10; |
|
|
|
|
|
data.key = ENC_KEY; |
|
|
|
|
|
data.model_data = new uint8_t[data.model_len]; |
|
|
|
|
|
uint8_t data_ori[model_len]; |
|
|
|
|
|
memset(data_ori, 0, model_len); |
|
|
|
|
|
ModelFileHeader *header = (ModelFileHeader *)data.model_data; |
|
|
|
|
|
header->magic = MODEL_FILE_MAGIC_NUM; |
|
|
|
|
|
header->version = MODEL_VERSION; |
|
|
|
|
|
header->is_encrypt = ModelEncryptType::ENCRYPTED; |
|
|
|
|
|
header->length = 10; // encrypt_len; |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
void LoadStandardModelData(ModelData &data) { |
|
|
void LoadStandardModelData(ModelData &data) { |
|
|
data.model_len = 512; |
|
|
data.model_len = 512; |
|
|
data.model_data = new uint8_t[data.model_len]; |
|
|
data.model_data = new uint8_t[data.model_len]; |
|
|