"vscode:/vscode.git/clone" did not exist on "2897f366a9a0a599409844d5a4bf8956c382512d"
Commit f773c763 authored by Khalique's avatar Khalique
Browse files

formatting

parent 4f685101
......@@ -498,7 +498,7 @@ struct onnx_parser
{
op::slice op;
std::vector<size_t> dims = args[0]->get_shape().lens();
size_t num_dims = dims.size();
size_t num_dims = dims.size();
if(contains(attributes, "axes"))
{
literal s = parse_value(attributes.at("axes"));
......@@ -509,12 +509,12 @@ struct onnx_parser
op.axes = std::vector<int64_t>(num_dims);
std::iota(op.axes.begin(), op.axes.end(), 0);
}
if(contains(attributes, "ends"))
{
literal s = parse_value(attributes.at("ends"));
s.visit([&](auto v) { copy(v, std::back_inserter(op.ends)); });
for (size_t i = 0; i < num_dims; i++)
for(size_t i = 0; i < num_dims; i++)
{
if(static_cast<size_t>(op.ends[i]) > dims[i])
{
......
......@@ -15,8 +15,7 @@ argument pad(hipStream_t stream,
argument result,
argument arg1,
float value,
std::vector<std::int64_t> pads
);
std::vector<std::int64_t> pads);
} // namespace device
} // namespace gpu
......
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