index 4072ac56eea8868d4e1257c6cbfb8cd0bc191c1b..cd9f87abe309ee59f7e2a63e91d066b9e4a75cf1 100644 (file)
}
+ // the components that we exclude from this test, are excluded because they
+ // all take "optional" right context, and this destroys the equivalence that
+ // we are testing.
if (!NnetIsRecurrent(*nnet) &&
nnet->Info().find("statistics-extraction") == std::string::npos &&
- nnet->Info().find("TimeHeightConvolutionComponent") == std::string::npos) {
+ nnet->Info().find("TimeHeightConvolutionComponent") == std::string::npos &&
+ nnet->Info().find("RestrictedAttentionComponent") == std::string::npos) {
// this equivalence will not hold for recurrent nnets, or those that
// have the statistics-extraction/statistics-pooling layers,
// or in general for nnets with convolution components (because these