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
|
||||
};
|
||||
|
||||
const std::vector<std::vector<size_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>> padsBegin2D = {{0, 0}, {1, 1}, {2, 0}, {0, 3}};
|
||||
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<ngraph::helpers::PadMode> padMode = {
|
||||
@ -57,8 +57,8 @@ INSTANTIATE_TEST_CASE_P(
|
||||
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<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>> padsBegin4D = {{0, 0, 0, 0}, {1, 1, 1, 1}, {2, 0, 1, 0}, {0, 3, 0, 1}};
|
||||
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(
|
||||
testing::ValuesIn(padsBegin4D),
|
||||
|
@ -13,8 +13,8 @@
|
||||
#include "ngraph_functions/builders.hpp"
|
||||
|
||||
typedef std::tuple<
|
||||
InferenceEngine::SizeVector, // padsBegin
|
||||
InferenceEngine::SizeVector, // padsEnd
|
||||
std::vector<int64_t>, // padsBegin
|
||||
std::vector<int64_t>, // padsEnd
|
||||
float, // argPadValue
|
||||
ngraph::helpers::PadMode, // padMode
|
||||
InferenceEngine::Precision, // Net precision
|
||||
|
@ -15,7 +15,8 @@ namespace LayerTestsDefinitions {
|
||||
|
||||
std::string PadLayerTest::getTestCaseName(testing::TestParamInfo<padLayerTestParamsSet> obj) {
|
||||
InferenceEngine::Precision netPrecision;
|
||||
InferenceEngine::SizeVector inputShapes, padsBegin, padsEnd;
|
||||
InferenceEngine::SizeVector inputShapes;
|
||||
std::vector<int64_t> padsBegin, padsEnd;
|
||||
ngraph::helpers::PadMode padMode;
|
||||
float argPadValue;
|
||||
std::string targetDevice;
|
||||
@ -35,7 +36,8 @@ std::string PadLayerTest::getTestCaseName(testing::TestParamInfo<padLayerTestPar
|
||||
}
|
||||
|
||||
void PadLayerTest::SetUp() {
|
||||
InferenceEngine::SizeVector inputShape, padsBegin, padsEnd;
|
||||
InferenceEngine::SizeVector inputShape;
|
||||
std::vector<int64_t> padsBegin, padsEnd;
|
||||
float argPadValue;
|
||||
ngraph::helpers::PadMode padMode;
|
||||
InferenceEngine::Precision netPrecision;
|
||||
|
@ -385,8 +385,8 @@ std::shared_ptr<ngraph::Node> makeConcat(const std::vector<ngraph::Output<Node>>
|
||||
const int& axis);
|
||||
|
||||
std::shared_ptr<ngraph::Node> makePad(const ngraph::Output<Node>& data,
|
||||
const std::vector<size_t>& padsBegin,
|
||||
const std::vector<size_t>& padsEnd,
|
||||
const std::vector<int64_t>& padsBegin,
|
||||
const std::vector<int64_t>& padsEnd,
|
||||
float argPadValue,
|
||||
ngraph::helpers::PadMode padMode);
|
||||
|
||||
|
@ -10,8 +10,8 @@
|
||||
namespace ngraph {
|
||||
namespace builder {
|
||||
std::shared_ptr<ngraph::Node> makePad(const ngraph::Output<Node>& data,
|
||||
const std::vector<size_t>& padsBegin,
|
||||
const std::vector<size_t>& padsEnd,
|
||||
const std::vector<int64_t>& padsBegin,
|
||||
const std::vector<int64_t>& padsEnd,
|
||||
float argPadValue,
|
||||
ngraph::helpers::PadMode padMode) {
|
||||
ngraph::op::PadMode pad_mode;
|
||||
|
Loading…
Reference in New Issue
Block a user