"examples/dequantize_gemm/utils.py" did not exist on "a86223f416a3f8183410d9fe38881b77fd536d48"
Commit d38e771c authored by Scott Thornton's avatar Scott Thornton
Browse files

Began adding transpose operator

parent 9af6974d
......@@ -54,6 +54,24 @@ struct cpu_convolution
}
};
struct cpu_transpose
{
transpose op;
std::string name() const { return "cpu::transpose"; }
shape compute_shape(std::vector<shape> inputs) const { return op.compute_shape(inputs); }
argument compute(shape output_shape, std::vector<argument> args) const
{
argument result{output_shape};
visit_all(result, args[0])([&](auto output, auto input) {
dfor(output_shape.lens()[0],
output_shape.lens()[0],
output_shape.lens()[0],
output_shape.lens()[0]
});
}
}
struct cpu_reshape
{
reshape op;
......
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