InferenceEngine::Precision::FP32
};
-const std::vector<std::vector<size_t>> repeats = {
+const std::vector<std::vector<int64_t>> repeats = {
{1, 2, 3},
{2, 1, 1},
{2, 3, 1},
INSTANTIATE_TEST_CASE_P(smoke_Tile6d, TileLayerTest,
::testing::Combine(
- ::testing::Values(std::vector<size_t>({1, 1, 1, 2, 1, 2})),
+ ::testing::Values(std::vector<int64_t>({1, 1, 1, 2, 1, 2})),
::testing::ValuesIn(netPrecisions),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
InferenceEngine::Precision::FP32
};
-const std::vector<std::vector<size_t>> repeats = {
+const std::vector<std::vector<int64_t>> repeats = {
{1, 2, 3},
{2, 1, 1},
{2, 3, 1},
INSTANTIATE_TEST_CASE_P(smoke_Tile6d, TileLayerTest,
::testing::Combine(
- ::testing::Values(std::vector<size_t>({1, 1, 1, 2, 1, 2})),
+ ::testing::Values(std::vector<int64_t>({1, 1, 1, 2, 1, 2})),
::testing::ValuesIn(netPrecisions),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
::testing::Values(InferenceEngine::Precision::UNSPECIFIED),
namespace LayerTestsDefinitions {
using batchToSpaceParamsTuple = typename std::tuple<
- std::vector<size_t>, // block shape
- std::vector<size_t>, // crops begin
- std::vector<size_t>, // crops end
+ std::vector<int64_t>, // block shape
+ std::vector<int64_t>, // crops begin
+ std::vector<int64_t>, // crops end
std::vector<size_t>, // Input shapes
InferenceEngine::Precision, // Network precision
InferenceEngine::Precision, // Input precision
namespace LayerTestsDefinitions {
using spaceToBatchParamsTuple = typename std::tuple<
- std::vector<size_t>, // block_shape
- std::vector<size_t>, // pads_begin
- std::vector<size_t>, // pads_end
+ std::vector<int64_t>, // block_shape
+ std::vector<int64_t>, // pads_begin
+ std::vector<int64_t>, // pads_end
std::vector<size_t>, // Input shapes
InferenceEngine::Precision, // Network precision
InferenceEngine::Precision, // Input precision
typedef std::tuple<
size_t, // Num splits
- size_t, // Axis
+ int64_t, // Axis
InferenceEngine::Precision, // Net precision
InferenceEngine::Precision, // Input precision
InferenceEngine::Precision, // Output precision
namespace LayerTestsDefinitions {
-typedef std::vector<size_t> TileSpecificParams;
+typedef std::vector<int64_t> TileSpecificParams;
typedef std::tuple<
TileSpecificParams,
InferenceEngine::Precision, // Net precision
namespace LayerTestsDefinitions {
std::string BatchToSpaceLayerTest::getTestCaseName(const testing::TestParamInfo<batchToSpaceParamsTuple> &obj) {
- std::vector<size_t> inShapes, blockShape, cropsBegin, cropsEnd;
+ std::vector<size_t> inShapes;
+ std::vector<int64_t> blockShape, cropsBegin, cropsEnd;
InferenceEngine::Precision netPrc;
InferenceEngine::Precision inPrc, outPrc;
InferenceEngine::Layout inLayout, outLayout;
void BatchToSpaceLayerTest::SetUp() {
SetRefMode(LayerTestsUtils::RefMode::INTERPRETER_TRANSFORMATIONS);
- std::vector<size_t> inputShape, blockShape, cropsBegin, cropsEnd;
+ std::vector<size_t> inputShape;
+ std::vector<int64_t> blockShape, cropsBegin, cropsEnd;
InferenceEngine::Precision netPrecision;
std::tie(blockShape, cropsBegin, cropsEnd, inputShape, netPrecision, inPrc, outPrc, inLayout, outLayout, targetDevice) = this->GetParam();
auto ngPrc = FuncTestUtils::PrecisionUtils::convertIE2nGraphPrc(netPrecision);
namespace LayerTestsDefinitions {
std::string SpaceToBatchLayerTest::getTestCaseName(const testing::TestParamInfo<spaceToBatchParamsTuple> &obj) {
- std::vector<size_t> inShapes, blockShape, padsBegin, padsEnd;
+ std::vector<size_t> inShapes;
+ std::vector<int64_t> blockShape, padsBegin, padsEnd;
InferenceEngine::Precision netPrc;
InferenceEngine::Precision inPrc, outPrc;
InferenceEngine::Layout inLayout, outLayout;
void SpaceToBatchLayerTest::SetUp() {
SetRefMode(LayerTestsUtils::RefMode::INTERPRETER_TRANSFORMATIONS);
- std::vector<size_t> inputShape, blockShape, padsBegin, padsEnd;
+ std::vector<size_t> inputShape;
+ std::vector<int64_t> blockShape, padsBegin, padsEnd;
InferenceEngine::Precision inputPrecision, netPrecision;
std::tie(blockShape, padsBegin, padsEnd, inputShape, netPrecision, inPrc, outPrc, inLayout, outLayout, targetDevice) = this->GetParam();
namespace LayerTestsDefinitions {
std::string SplitLayerTest::getTestCaseName(testing::TestParamInfo<splitParams> obj) {
- size_t numSplits, axis;
+ size_t numSplits;
+ int64_t axis;
InferenceEngine::Precision netPrecision;
InferenceEngine::Precision inPrc, outPrc;
InferenceEngine::Layout inLayout, outLayout;
std::shared_ptr<ngraph::Node> makeSplit(const ngraph::Output<Node> &in,
const element::Type &type,
size_t numSplits,
- size_t axis);
+ int64_t axis);
std::shared_ptr<ngraph::Node> makeVariadicSplit(const ngraph::Output<Node> &in,
const std::vector<size_t> numSplits,
std::shared_ptr<ngraph::Node> makeBatchToSpace(const ngraph::Output<Node> &in,
const element::Type &type,
- const std::vector<size_t> &blockShape,
- const std::vector<size_t> &cropsBegin,
- const std::vector<size_t> &cropsEnd);
+ const std::vector<int64_t> &blockShape,
+ const std::vector<int64_t> &cropsBegin,
+ const std::vector<int64_t> &cropsEnd);
std::shared_ptr<ngraph::Node> makeSpaceToBatch(const ngraph::Output<Node> &in,
const element::Type &type,
- const std::vector<size_t> &blockShape,
- const std::vector<size_t> &padsBegin,
- const std::vector<size_t> &padsEnd);
+ const std::vector<int64_t> &blockShape,
+ const std::vector<int64_t> &padsBegin,
+ const std::vector<int64_t> &padsEnd);
std::shared_ptr<ngraph::Node> makeStridedSlice(const ngraph::Output<Node> &in,
const std::vector<int64_t> &begin,
const std::size_t batchDims);
std::shared_ptr<ngraph::Node> makeTile(const ngraph::Output<Node>& in,
- const std::vector<size_t>& repeats);
+ const std::vector<int64_t>& repeats);
std::shared_ptr<ngraph::Node> makeNormalizeL2(const ngraph::Output<Node>& data,
const std::vector<int64_t>& axes,
namespace builder {
std::shared_ptr<ngraph::Node> makeBatchToSpace(const ngraph::Output<Node> &in,
const element::Type &type,
- const std::vector<size_t> &blockShape,
- const std::vector<size_t> &cropsBegin,
- const std::vector<size_t> &cropsEnd) {
+ const std::vector<int64_t> &blockShape,
+ const std::vector<int64_t> &cropsBegin,
+ const std::vector<int64_t> &cropsEnd) {
ngraph::Shape constShape = {in.get_shape().size()};
auto blockShapeNode = std::make_shared<ngraph::opset1::Constant>(ngraph::element::i64, constShape,
blockShape.data());
namespace builder {
std::shared_ptr<ngraph::Node> makeSpaceToBatch(const ngraph::Output<Node> &in,
const element::Type &type,
- const std::vector<size_t> &blockShape,
- const std::vector<size_t> &padsBegin,
- const std::vector<size_t> &padsEnd) {
+ const std::vector<int64_t> &blockShape,
+ const std::vector<int64_t> &padsBegin,
+ const std::vector<int64_t> &padsEnd) {
ngraph::Shape constShape = {in.get_shape().size()};
auto blockShapeNode = std::make_shared<ngraph::opset1::Constant>(ngraph::element::i64, constShape,
blockShape.data());
std::shared_ptr<ngraph::Node> makeSplit(const ngraph::Output<Node> &in,
const element::Type &type,
size_t numSplits,
- size_t axis) {
+ int64_t axis) {
auto splitAxisOp = std::make_shared<ngraph::opset1::Constant>(ngraph::element::Type_t::i64, ngraph::Shape{},
- std::vector<uint64_t>{axis});
+ std::vector<int64_t>{axis});
auto splitNode = std::make_shared<ngraph::opset1::Split>(in, splitAxisOp, numSplits);
return splitNode;
}
namespace builder {
std::shared_ptr<ngraph::Node> makeTile(const ngraph::Output<Node>& in,
- const std::vector<size_t>& repeats) {
+ const std::vector<int64_t>& repeats) {
auto repeatsNode = std::make_shared<ngraph::opset1::Constant>(ngraph::element::i64, std::vector<size_t>{repeats.size()}, repeats);
auto tileNode = std::make_shared<ngraph::opset1::Tile>(in, repeatsNode);
return tileNode;