Change-Id: Iec43db205d77dc9864abe1fe85104722a6954d57
Signed-off-by: Ruomei Yan <ruomei.yan@arm.com>
CHECK_VALID_SIZE(outputs.size(), 1);
auto outputInfo = ToTensorInfo(outputs[0]);
- auto layerName = boost::str(boost::format("BatchNormalization:%1%") % layerIndex);
+ auto layerName = GetLayerName(graph, layerIndex);
auto serializerLayer = graph->layers()->Get(layerIndex)->layer_as_BatchNormalizationLayer();
auto serializerDescriptor = serializerLayer->descriptor();
const armnn::ConstTensor&,
const char* name) override
{
- BOOST_TEST(name == "BatchNormalization:1");
+ BOOST_TEST(name == "BatchNormalizationLayer");
}
};