Irina Efode
602f8f2e08
* [IE TESTS] Changing functional test utils structure * Example * Remove extra * Apply comments * fixes * [IE TESTS] Change the structure * Continue * step 3 * [IE TESTS] Complete transition single layer test classes * [IE TESTS] Transition Subgraph * Fix subgraph namespaces * fix * Apply comments * latm fix
57 lines
2.0 KiB
C++
57 lines
2.0 KiB
C++
// Copyright (C) 2020 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#include <shared_test_classes/single_layer/convert.hpp>
|
|
|
|
using namespace LayerTestsDefinitions;
|
|
using namespace InferenceEngine;
|
|
|
|
namespace CPULayerTestsDefinitions {
|
|
|
|
class ConvertCPULayerTest : public ConvertLayerTest {};
|
|
|
|
TEST_P(ConvertCPULayerTest, CompareWithRefs) {
|
|
SKIP_IF_CURRENT_TEST_IS_DISABLED()
|
|
|
|
ConvertParamsTuple params = GetParam();
|
|
inPrc = std::get<1>(params);
|
|
outPrc = std::get<2>(params);
|
|
|
|
Run();
|
|
}
|
|
|
|
namespace {
|
|
const std::vector<std::vector<size_t>> inShape = {{1, 2, 3, 4}};
|
|
|
|
// List of precisions natively supported by mkldnn.
|
|
const std::vector<Precision> precisions = {
|
|
Precision::U8,
|
|
Precision::I8,
|
|
Precision::I16,
|
|
Precision::I32,
|
|
Precision::FP32,
|
|
Precision::BF16
|
|
};
|
|
|
|
INSTANTIATE_TEST_CASE_P(smoke_ConvertLayerTest_From_BF16, ConvertCPULayerTest,
|
|
::testing::Combine(
|
|
::testing::Values(inShape),
|
|
::testing::Values(Precision::BF16),
|
|
::testing::ValuesIn(precisions),
|
|
::testing::Values(Layout::ANY),
|
|
::testing::Values(Layout::ANY),
|
|
::testing::Values(CommonTestUtils::DEVICE_CPU)),
|
|
ConvertLayerTest::getTestCaseName);
|
|
|
|
INSTANTIATE_TEST_CASE_P(smoke_ConvertLayerTest_To_BF16, ConvertCPULayerTest,
|
|
::testing::Combine(
|
|
::testing::Values(inShape),
|
|
::testing::ValuesIn(precisions),
|
|
::testing::Values(Precision::BF16),
|
|
::testing::Values(Layout::ANY),
|
|
::testing::Values(Layout::ANY),
|
|
::testing::Values(CommonTestUtils::DEVICE_CPU)),
|
|
ConvertLayerTest::getTestCaseName);
|
|
} // namespace
|
|
} // namespace CPULayerTestsDefinitions
|