Commit 5a0c8ca7 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

correct a typo

parent cea3aa4b
...@@ -19,19 +19,19 @@ namespace op { ...@@ -19,19 +19,19 @@ namespace op {
struct convert struct convert
{ {
shape::type_t targe_type = shape::half_type; shape::type_t target_type = shape::half_type;
template <class Self, class F> template <class Self, class F>
static auto reflect(Self& self, F f) static auto reflect(Self& self, F f)
{ {
return pack(f(self.targe_type, "target_type")); return pack(f(self.target_type, "target_type"));
} }
std::string name() const { return "convert"; } std::string name() const { return "convert"; }
shape compute_shape(std::vector<shape> inputs) const shape compute_shape(std::vector<shape> inputs) const
{ {
check_shapes{inputs, *this}.has(1); check_shapes{inputs, *this}.has(1);
return {targe_type, inputs.front().lens(), inputs.front().strides()}; return {target_type, inputs.front().lens(), inputs.front().strides()};
} }
argument compute(const shape& output_shape, std::vector<argument> args) const argument compute(const shape& output_shape, std::vector<argument> args) 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