提交 e0f0f98f 编写于 作者: L liuqi

Fix std::to_string bug in concat test.

上级 0608836d
...@@ -97,7 +97,7 @@ TEST_F(ConcatOpTest, Random) { ...@@ -97,7 +97,7 @@ TEST_F(ConcatOpTest, Random) {
auto &net = test_net(); auto &net = test_net();
auto builder = OpDefBuilder("Concat", "ConcatTest"); auto builder = OpDefBuilder("Concat", "ConcatTest");
for (int i = 0; i < num_inputs; ++i) { for (int i = 0; i < num_inputs; ++i) {
builder = builder.Input(("Input" + std::to_string(i)).c_str()); builder = builder.Input(("Input" + ToString(i)).c_str());
} }
builder.Input("Axis") builder.Input("Axis")
.Output("Output") .Output("Output")
...@@ -114,7 +114,7 @@ TEST_F(ConcatOpTest, Random) { ...@@ -114,7 +114,7 @@ TEST_F(ConcatOpTest, Random) {
concat_axis_size += input_shapes[i][axis]; concat_axis_size += input_shapes[i][axis];
GenerateRandomRealTypeData(input_shapes[i], inputs[i]); GenerateRandomRealTypeData(input_shapes[i], inputs[i]);
input_ptrs[i] = inputs[i].data(); input_ptrs[i] = inputs[i].data();
net.AddInputFromArray<float>(("Input" + std::to_string(i)).c_str(), input_shapes[i], inputs[i]); net.AddInputFromArray<float>(("Input" + ToString(i)).c_str(), input_shapes[i], inputs[i]);
} }
net.AddInputFromArray<int>("Axis", {}, {axis}); net.AddInputFromArray<int>("Axis", {}, {axis});
......
...@@ -302,6 +302,13 @@ void ExpectTensorNear(const Tensor &x, const Tensor &y, const double abs_err) { ...@@ -302,6 +302,13 @@ void ExpectTensorNear(const Tensor &x, const Tensor &y, const double abs_err) {
Expector<T>::Near(x, y, abs_err); Expector<T>::Near(x, y, abs_err);
} }
template<typename T>
std::string ToString(const T& input) {
std::stringstream ss;
ss<<input;
return ss.str();
}
} // namespace mace } // namespace mace
#endif // MACE_OPS_TEST_UTIL_H_ #endif // MACE_OPS_TEST_UTIL_H_
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册