"vscode:/vscode.git/clone" did not exist on "a9dcc00670c611f8d610213d12ca2e54099a3913"
Commit 915ecaa5 authored by Paul's avatar Paul
Browse files

Format

parent c310d928
...@@ -103,11 +103,11 @@ struct fused_reduce_compiler : compiler<fused_reduce_compiler> ...@@ -103,11 +103,11 @@ struct fused_reduce_compiler : compiler<fused_reduce_compiler>
operation compile_op(context& ctx, const std::vector<shape>& inputs, const value& v) const operation compile_op(context& ctx, const std::vector<shape>& inputs, const value& v) const
{ {
auto axes = v.at("axes").to_vector<std::size_t>(); auto axes = v.at("axes").to_vector<std::size_t>();
auto virtual_inputs = inputs; auto virtual_inputs = inputs;
virtual_inputs.push_back(get_reduced_shape(inputs.front(), axes)); virtual_inputs.push_back(get_reduced_shape(inputs.front(), axes));
virtual_inputs.push_back(get_output_shape(inputs.front(), axes)); virtual_inputs.push_back(get_output_shape(inputs.front(), axes));
virtual_inputs = reduce_dims(virtual_inputs); virtual_inputs = reduce_dims(virtual_inputs);
auto output_shape = virtual_inputs.back(); auto output_shape = virtual_inputs.back();
virtual_inputs.pop_back(); virtual_inputs.pop_back();
auto reduced_shape = virtual_inputs.back(); auto reduced_shape = virtual_inputs.back();
......
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