a946079208
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix train compile

This commit is contained in:
lz 2020-10-13 10:42:59 +08:00
parent afa5ed3e5c
commit a2f0e7fd16
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/pooling_grad_fp32_tests.cc
View File

@ -16,6 +16,7 @@
#include <iostream>
#include <memory>
#include "src/ops/primitive_c.h"
#include "mindspore/lite/include/context.h"
#include "utils/log_adapter.h"
#include "common/common_test.h"

5
mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/softmax_grad_fp32_tests.cc
View File

@ -17,10 +17,7 @@
#include <iostream>
#include <memory>
#include <vector>
// #include "mindspore/lite/src/ir/tensor.h"
// #include "mindspore/lite/src/lite_kernel.h"
#include "src/ops/primitive_c.h"
#include "mindspore/lite/include/context.h"
#include "utils/log_adapter.h"
#include "common/common_test.h"