|
|
|
@ -991,8 +991,10 @@ TEST(Layer, SequenceLastInstanceLayer) {
|
|
|
|
|
"seqlastins",
|
|
|
|
|
"non-seq",
|
|
|
|
|
-1); // hasSubseq seqlastins to non-seq
|
|
|
|
|
testDegradeLayer(
|
|
|
|
|
true, "seqlastins", "seq", -1); // hasSubseq seqlastins to seq
|
|
|
|
|
testDegradeLayer(true,
|
|
|
|
|
"seqlastins",
|
|
|
|
|
"seq",
|
|
|
|
|
-1); // hasSubseq seqlastins to seq
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
TEST(Layer, AverageLayer) {
|
|
|
|
@ -1001,8 +1003,10 @@ TEST(Layer, AverageLayer) {
|
|
|
|
|
"average",
|
|
|
|
|
"non-seq",
|
|
|
|
|
5); // seq average to a shorten seq, stride window = 5
|
|
|
|
|
testDegradeLayer(
|
|
|
|
|
true, "average", "non-seq", -1); // hasSubseq average to non-seq
|
|
|
|
|
testDegradeLayer(true,
|
|
|
|
|
"average",
|
|
|
|
|
"non-seq",
|
|
|
|
|
-1); // hasSubseq average to non-seq
|
|
|
|
|
testDegradeLayer(true, "average", "seq", -1); // hasSubseq average to seq
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -1287,8 +1291,9 @@ TEST(Layer, PoolLayer) {
|
|
|
|
|
testPoolLayer("cudnn-avg-pool", /* trans= */ false, /* useGpu= */ true);
|
|
|
|
|
testPoolLayer2("cudnn-max-pool", /* trans= */ false, /* useGpu= */ true);
|
|
|
|
|
testPoolLayer2("cudnn-avg-pool", /* trans= */ false, /* useGpu= */ true);
|
|
|
|
|
testPoolLayer2(
|
|
|
|
|
"cudnn-avg-incl-pad-pool", /* trans= */ false, /* useGpu= */ true);
|
|
|
|
|
testPoolLayer2("cudnn-avg-incl-pad-pool",
|
|
|
|
|
/* trans= */ false,
|
|
|
|
|
/* useGpu= */ true);
|
|
|
|
|
testPoolLayer("max-pool-with-mask", /* trans= */ false, /* useGpu= */ true);
|
|
|
|
|
#endif
|
|
|
|
|
}
|
|
|
|
@ -2431,18 +2436,21 @@ TEST(Layer, test3DDeConvLayer) {
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
TEST(Layer, ScaleShiftLayer) {
|
|
|
|
|
const size_t batchSize = 16;
|
|
|
|
|
const size_t size = 32;
|
|
|
|
|
TestConfig config;
|
|
|
|
|
config.layerConfig.set_type("scale_shift");
|
|
|
|
|
config.layerConfig.set_size(size);
|
|
|
|
|
config.biasSize = 1;
|
|
|
|
|
config.inputDefs.push_back(
|
|
|
|
|
{INPUT_DATA, "input", /* dim= */ size, /* paraSize= */ 1});
|
|
|
|
|
config.layerConfig.add_inputs();
|
|
|
|
|
for (auto useGpu : {false, true}) {
|
|
|
|
|
testLayerGrad(config, "scale_shift", batchSize, false, useGpu, false);
|
|
|
|
|
}
|
|
|
|
|
// FIXME: Disable ScaleShiftLayer because it is not stable.
|
|
|
|
|
// https://github.com/PaddlePaddle/Paddle/issues/7781
|
|
|
|
|
return;
|
|
|
|
|
// const size_t batchSize = 16;
|
|
|
|
|
// const size_t size = 32;
|
|
|
|
|
// TestConfig config;
|
|
|
|
|
// config.layerConfig.set_type("scale_shift");
|
|
|
|
|
// config.layerConfig.set_size(size);
|
|
|
|
|
// config.biasSize = 1;
|
|
|
|
|
// config.inputDefs.push_back(
|
|
|
|
|
// {INPUT_DATA, "input", /* dim= */ size, /* paraSize= */ 1});
|
|
|
|
|
// config.layerConfig.add_inputs();
|
|
|
|
|
// for (auto useGpu : {false, true}) {
|
|
|
|
|
// testLayerGrad(config, "scale_shift", batchSize, false, useGpu, false);
|
|
|
|
|
// }
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
TEST(Layer, ScaleSubRegionLayer) {
|
|
|
|
|