"src/vscode:/vscode.git/clone" did not exist on "d8756a4eb66bb1cb1c9350271fb3bc3f900e81f2"
Commit cbf4c8d6 authored by Paul's avatar Paul
Browse files

More formatting

parent 8f330074
......@@ -238,17 +238,20 @@ struct transpose
auto input_lens = input.lens();
auto input_strides = input.strides();
auto t = input.type();
if (dims.size() != input_lens.size()) {
if(dims.size() != input_lens.size())
{
RTG_THROW("Permutation has wrong number of axes");
}
std::vector<int64_t> axes(dims.size());
std::iota(axes.begin(), axes.end(), 0);
if (!std::is_permutation(axes.begin(), axes.end(), dims.begin())) {
if(!std::is_permutation(axes.begin(), axes.end(), dims.begin()))
{
RTG_THROW("Invalid permutation");
}
std::vector<size_t> output_lens(input_lens.size());
std::vector<size_t> output_strides(input_lens.size());
for (int i = 0; i < output_lens.size(); i++) {
for(int i = 0; i < output_lens.size(); i++)
{
output_lens[i] = input_lens[dims[i]];
output_strides[i] = input_strides[dims[i]];
}
......@@ -265,7 +268,8 @@ struct contiguous
check_shapes{inputs}.has(1);
auto lens = inputs.at(0).lens();
auto t = inputs.at(0).type();
if (lens.size() < 2) {
if(lens.size() < 2)
{
RTG_THROW("Number of dimensions should exceed 1");
}
return {t, lens};
......
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