Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
Browse Source

fix compile training

tags/v1.1.0
lz 5 years ago
parent
39bc43e674
commit
f6fd4fe3c0
2 changed files with 2 additions and 4 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -0
      mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/pooling_grad_fp32_tests.cc
  2. +1
    -4
      mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/softmax_grad_fp32_tests.cc

+ 1
- 0
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 "src/common/log_adapter.h"
#include "common/common_test.h"


+ 1
- 4
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 "src/common/log_adapter.h"
#include "common/common_test.h"


Write Preview
Loading…
Cancel
Save