Changed pad data type (#2354)
This commit is contained in:
parent
101ef50f4e
commit
f8a17a1317
@ -14,8 +14,8 @@ const std::vector<InferenceEngine::Precision> netPrecisions = {
|
|||||||
InferenceEngine::Precision::FP16
|
InferenceEngine::Precision::FP16
|
||||||
};
|
};
|
||||||
|
|
||||||
const std::vector<std::vector<size_t>> padsBegin2D = {{0, 0}, {1, 1}, {2, 0}, {0, 3}};
|
const std::vector<std::vector<int64_t>> padsBegin2D = {{0, 0}, {1, 1}, {2, 0}, {0, 3}};
|
||||||
const std::vector<std::vector<size_t>> padsEnd2D = {{0, 0}, {1, 1}, {0, 1}, {3, 2}};
|
const std::vector<std::vector<int64_t>> padsEnd2D = {{0, 0}, {1, 1}, {0, 1}, {3, 2}};
|
||||||
const std::vector<float> argPadValue = {0.f, 1.f, 2.f, -1.f};
|
const std::vector<float> argPadValue = {0.f, 1.f, 2.f, -1.f};
|
||||||
|
|
||||||
const std::vector<ngraph::helpers::PadMode> padMode = {
|
const std::vector<ngraph::helpers::PadMode> padMode = {
|
||||||
@ -57,8 +57,8 @@ INSTANTIATE_TEST_CASE_P(
|
|||||||
PadLayerTest::getTestCaseName
|
PadLayerTest::getTestCaseName
|
||||||
);
|
);
|
||||||
|
|
||||||
const std::vector<std::vector<size_t>> padsBegin4D = {{0, 0, 0, 0}, {1, 1, 1, 1}, {2, 0, 1, 0}, {0, 3, 0, 1}};
|
const std::vector<std::vector<int64_t>> padsBegin4D = {{0, 0, 0, 0}, {1, 1, 1, 1}, {2, 0, 1, 0}, {0, 3, 0, 1}};
|
||||||
const std::vector<std::vector<size_t>> padsEnd4D = {{0, 0, 0, 0}, {1, 1, 1, 1}, {2, 0, 0, 1}, {1, 3, 2, 0}};
|
const std::vector<std::vector<int64_t>> padsEnd4D = {{0, 0, 0, 0}, {1, 1, 1, 1}, {2, 0, 0, 1}, {1, 3, 2, 0}};
|
||||||
|
|
||||||
const auto pad4DConstparams = testing::Combine(
|
const auto pad4DConstparams = testing::Combine(
|
||||||
testing::ValuesIn(padsBegin4D),
|
testing::ValuesIn(padsBegin4D),
|
||||||
|
@ -13,8 +13,8 @@
|
|||||||
#include "ngraph_functions/builders.hpp"
|
#include "ngraph_functions/builders.hpp"
|
||||||
|
|
||||||
typedef std::tuple<
|
typedef std::tuple<
|
||||||
InferenceEngine::SizeVector, // padsBegin
|
std::vector<int64_t>, // padsBegin
|
||||||
InferenceEngine::SizeVector, // padsEnd
|
std::vector<int64_t>, // padsEnd
|
||||||
float, // argPadValue
|
float, // argPadValue
|
||||||
ngraph::helpers::PadMode, // padMode
|
ngraph::helpers::PadMode, // padMode
|
||||||
InferenceEngine::Precision, // Net precision
|
InferenceEngine::Precision, // Net precision
|
||||||
|
@ -15,7 +15,8 @@ namespace LayerTestsDefinitions {
|
|||||||
|
|
||||||
std::string PadLayerTest::getTestCaseName(testing::TestParamInfo<padLayerTestParamsSet> obj) {
|
std::string PadLayerTest::getTestCaseName(testing::TestParamInfo<padLayerTestParamsSet> obj) {
|
||||||
InferenceEngine::Precision netPrecision;
|
InferenceEngine::Precision netPrecision;
|
||||||
InferenceEngine::SizeVector inputShapes, padsBegin, padsEnd;
|
InferenceEngine::SizeVector inputShapes;
|
||||||
|
std::vector<int64_t> padsBegin, padsEnd;
|
||||||
ngraph::helpers::PadMode padMode;
|
ngraph::helpers::PadMode padMode;
|
||||||
float argPadValue;
|
float argPadValue;
|
||||||
std::string targetDevice;
|
std::string targetDevice;
|
||||||
@ -35,7 +36,8 @@ std::string PadLayerTest::getTestCaseName(testing::TestParamInfo<padLayerTestPar
|
|||||||
}
|
}
|
||||||
|
|
||||||
void PadLayerTest::SetUp() {
|
void PadLayerTest::SetUp() {
|
||||||
InferenceEngine::SizeVector inputShape, padsBegin, padsEnd;
|
InferenceEngine::SizeVector inputShape;
|
||||||
|
std::vector<int64_t> padsBegin, padsEnd;
|
||||||
float argPadValue;
|
float argPadValue;
|
||||||
ngraph::helpers::PadMode padMode;
|
ngraph::helpers::PadMode padMode;
|
||||||
InferenceEngine::Precision netPrecision;
|
InferenceEngine::Precision netPrecision;
|
||||||
|
@ -385,8 +385,8 @@ std::shared_ptr<ngraph::Node> makeConcat(const std::vector<ngraph::Output<Node>>
|
|||||||
const int& axis);
|
const int& axis);
|
||||||
|
|
||||||
std::shared_ptr<ngraph::Node> makePad(const ngraph::Output<Node>& data,
|
std::shared_ptr<ngraph::Node> makePad(const ngraph::Output<Node>& data,
|
||||||
const std::vector<size_t>& padsBegin,
|
const std::vector<int64_t>& padsBegin,
|
||||||
const std::vector<size_t>& padsEnd,
|
const std::vector<int64_t>& padsEnd,
|
||||||
float argPadValue,
|
float argPadValue,
|
||||||
ngraph::helpers::PadMode padMode);
|
ngraph::helpers::PadMode padMode);
|
||||||
|
|
||||||
|
@ -10,8 +10,8 @@
|
|||||||
namespace ngraph {
|
namespace ngraph {
|
||||||
namespace builder {
|
namespace builder {
|
||||||
std::shared_ptr<ngraph::Node> makePad(const ngraph::Output<Node>& data,
|
std::shared_ptr<ngraph::Node> makePad(const ngraph::Output<Node>& data,
|
||||||
const std::vector<size_t>& padsBegin,
|
const std::vector<int64_t>& padsBegin,
|
||||||
const std::vector<size_t>& padsEnd,
|
const std::vector<int64_t>& padsEnd,
|
||||||
float argPadValue,
|
float argPadValue,
|
||||||
ngraph::helpers::PadMode padMode) {
|
ngraph::helpers::PadMode padMode) {
|
||||||
ngraph::op::PadMode pad_mode;
|
ngraph::op::PadMode pad_mode;
|
||||||
|
Loading…
Reference in New Issue
Block a user