Commit fa542e7c authored by Shucai Xiao's avatar Shucai Xiao
Browse files

fixed review comments

parent f426922a
...@@ -10,7 +10,7 @@ inline namespace MIGRAPHX_INLINE_NS { ...@@ -10,7 +10,7 @@ inline namespace MIGRAPHX_INLINE_NS {
void auto_contiguous::apply(module& p) const void auto_contiguous::apply(module& p) const
{ {
std::string key = "standard_input_shape"; std::string key = "std_shape";
for(auto ins : reverse_iterator_for(p)) for(auto ins : reverse_iterator_for(p))
{ {
auto&& attr = ins->get_operator().attributes(); auto&& attr = ins->get_operator().attributes();
......
...@@ -66,7 +66,7 @@ struct reduce_op : op_name<Derived> ...@@ -66,7 +66,7 @@ struct reduce_op : op_name<Derived>
{ {
value normalize; value normalize;
normalize["axes"] = value::array{normalize_attribute::include_min}; normalize["axes"] = value::array{normalize_attribute::include_min};
return {{"normalize_axes", normalize}, {"standard_input_shape", true}}; return {{"normalize_axes", normalize}, {"std_shape", true}};
} }
std::vector<int64_t> tune_axes(std::size_t n_dim) const std::vector<int64_t> tune_axes(std::size_t n_dim) const
......
...@@ -27,7 +27,7 @@ struct reshape ...@@ -27,7 +27,7 @@ struct reshape
return pack(f(self.dims, "dims")); return pack(f(self.dims, "dims"));
} }
value attributes() const { return {{"standard_input_shape", true}}; } value attributes() const { return {{"std_shape", true}}; }
std::string name() const { return "reshape"; } std::string name() const { return "reshape"; }
shape compute_shape(std::vector<shape> inputs) const shape compute_shape(std::vector<shape> inputs) const
......
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