saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-14 20:48:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Accomodating PR comments.

Browse source
This commit is contained in:
Du Li 2018-11-26 21:02:18 -08:00
parent e7e801b45e
commit 0bb5cb7a30
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/test/contrib_ops/expand_dims_test.cc
View file

@ -9,7 +9,7 @@ namespace test {
TEST(ContribOpTest, ExpandDims_0) {
OpTester test("ExpandDims", 1, onnxruntime::kMSDomain);
test.AddShapeToTensorData(true); // TODO: re-enable shape inference test
test.AddShapeToTensorData(true);
test.AddInput<float>("X", {2, 3}, std::vector<float>(6, 1.0f));
test.AddInput<int32_t>("axis", {}, {-1});
test.AddOutput<float>("Y", {2, 3, 1}, std::vector<float>(6, 1.0f));
@ -18,7 +18,7 @@ TEST(ContribOpTest, ExpandDims_0) {
TEST(ContribOpTest, ExpandDims_1) {
OpTester test("ExpandDims", 1, onnxruntime::kMSDomain);
test.AddShapeToTensorData(true); // TODO: re-enable shape inference test
test.AddShapeToTensorData(true);
test.AddInput<float>("X", {2, 3}, std::vector<float>(6, 1.0f));
test.AddInput<int32_t>("axis", {}, {1});
test.AddOutput<float>("Y", {2, 1, 3}, std::vector<float>(6, 1.0f));