Commit e5de78ae authored by Khalique Ahmed's avatar Khalique Ahmed
Browse files

formatting

parent bca45a1a
...@@ -135,7 +135,8 @@ struct convolution ...@@ -135,7 +135,8 @@ struct convolution
ret.push_back(std::size_t(std::max<std::ptrdiff_t>( ret.push_back(std::size_t(std::max<std::ptrdiff_t>(
1, 1,
(x_lens[i + 1] - (1 + dilation[i] * (*(w_lens.rbegin() + i) - 1)) + padding_factor) / (x_lens[i + 1] - (1 + dilation[i] * (*(w_lens.rbegin() + i) - 1)) +
padding_factor) /
stride[i] + stride[i] +
1))); 1)));
} }
......
...@@ -138,9 +138,9 @@ inline tensor_descriptor make_tensor(const migraphx::shape& os, bool pack = fals ...@@ -138,9 +138,9 @@ inline tensor_descriptor make_tensor(const migraphx::shape& os, bool pack = fals
MIGRAPHX_THROW("MAKE_TENSOR: unsupported type"); MIGRAPHX_THROW("MAKE_TENSOR: unsupported type");
} }
// miopenSetTensorDescriptor(t.get(), d, s.lens().size(), lens.data(), strides.data()); // miopenSetTensorDescriptor(t.get(), d, s.lens().size(), lens.data(), strides.data());
miopenSetNdTensorDescriptorWithLayout(t.get(), d, miopenTensorLayout_t::miopenTensorNHWC, lens.data(), lens.size()); miopenSetNdTensorDescriptorWithLayout(
t.get(), d, miopenTensorLayout_t::miopenTensorNHWC, lens.data(), lens.size());
return t; return t;
} }
......
...@@ -60,8 +60,7 @@ struct parse_pack : op_parser<parse_pack> ...@@ -60,8 +60,7 @@ struct parse_pack : op_parser<parse_pack>
[&](instruction_ref arg) { [&](instruction_ref arg) {
return info.add_instruction(make_op("unsqueeze", {{"axes", {axis}}}), arg); return info.add_instruction(make_op("unsqueeze", {{"axes", {axis}}}), arg);
}); });
return return info.add_instruction(make_op("concat", {{"axis", axis}}), unsqueezed_args);
info.add_instruction(make_op("concat", {{"axis", axis}}), unsqueezed_args);
// return parser.to_nhwc( // return parser.to_nhwc(
// info.add_instruction(make_op("concat", {{"axis", axis}}), unsqueezed_args)); // info.add_instruction(make_op("concat", {{"axis", axis}}), unsqueezed_args));
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment