File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed
Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -15,8 +15,10 @@ limitations under the License. */
1515#include < gtest/gtest.h>
1616#include < string>
1717#include < vector>
18+ #include " ./paddle/utils/CommandLineParser.h"
1819#include " ModelConfig.pb.h"
1920#include " paddle/gserver/layers/DataLayer.h"
21+ #include " paddle/gserver/layers/ExpandConvTransLayer.h"
2022#include " paddle/math/MathUtils.h"
2123#include " paddle/trainer/Trainer.h"
2224#include " paddle/utils/GlobalConstants.h"
@@ -29,7 +31,9 @@ using namespace std; // NOLINT
2931
3032P_DECLARE_bool (use_gpu);
3133P_DECLARE_int32 (gpu_id);
34+ P_DECLARE_double (checkgrad_eps);
3235P_DECLARE_bool (thread_local_rand_use_global_seed);
36+ P_DECLARE_bool (prev_batch_state);
3337
3438// Do one forward pass of priorBox layer and check to see if its output
3539// matches the given result
You can’t perform that action at this time.
0 commit comments