[CPU] Add explicit storage for MemoryNode (#895)

This commit is contained in:
Alexander Peskov
2020-08-13 19:06:20 +03:00
committed by GitHub
parent 9a62e00674
commit 7c921b8b45
26 changed files with 1004 additions and 49 deletions

View File

@@ -271,7 +271,7 @@ class EltwiseOnlyTest : public TestsCommon,
if (p.op != eltwise_test_params::Pow)
CommonTestUtils::fill_data_sine(inputBlob->buffer().as<float*>(), inputBlob->size(), 100, 10, 10);
else
CommonTestUtils::fill_data_const(inputBlob->buffer().as<float*>(), inputBlob->size(), 2);
CommonTestUtils::fill_data_const(inputBlob, 2);
srcs_vec.push_back(inputBlob);
}

View File

@@ -240,7 +240,7 @@ protected:
Blob::Ptr output_low_data = make_shared_blob<float>({Precision::FP32, { p.ic_const_blobs }, Layout::C});
output_low_data->allocate();
if (p.levels == 2) {
CommonTestUtils::fill_data_const(output_low_data->buffer().as<float*>(), output_low_data->size(), low_val);
CommonTestUtils::fill_data_const(output_low_data, low_val);
} else {
CommonTestUtils::fill_data_sine(output_low_data->buffer().as<float*>(), output_low_data->size(), low_center, 2.f, 0.3f);
};
@@ -249,7 +249,7 @@ protected:
Blob::Ptr output_high_data = make_shared_blob<float>({Precision::FP32, {p.ic_const_blobs}, Layout::C});
output_high_data->allocate();
if (p.levels == 2) {
CommonTestUtils::fill_data_const(output_high_data->buffer().as<float*>(), output_high_data->size(), high_val);
CommonTestUtils::fill_data_const(output_high_data, high_val);
} else {
CommonTestUtils::fill_data_sine(output_high_data->buffer().as<float*>(), output_high_data->size(), high_center, 2.f, 0.3f);
};