提交 01c3c0e8 编写于 作者: W Wiktor Adamski

Code style changes.

上级 6ddd532d
...@@ -98,34 +98,33 @@ class OneHotOp<DeviceType::CPU, T> : public OneHotOpBase { ...@@ -98,34 +98,33 @@ class OneHotOp<DeviceType::CPU, T> : public OneHotOpBase {
} }
} }
} else { } else {
run(input, input_ptr, output_ptr, axis, 0, 0, input_shape.size(), 0); run(input, &input_ptr, &output_ptr, axis, 0, 0, input_shape.size(), 0);
} }
return MaceStatus::MACE_SUCCESS; return MaceStatus::MACE_SUCCESS;
} }
private: private:
void run(const Tensor *input, const T *&input_ptr, void run(const Tensor *input, const T **input_ptr,
T *&output_ptr, const index_t axis, T **output_ptr, const index_t axis,
const index_t current_in, const index_t current_out, const index_t current_in, const index_t current_out,
const index_t left, const index_t test) const { const index_t left, const index_t test) const {
if (current_out == axis) { if (current_out == axis) {
const index_t length = depth_; const index_t length = depth_;
if (left == 0) { if (left == 0) {
for (index_t i = 0; i < length; ++i) { for (index_t i = 0; i < length; ++i) {
*output_ptr = *input_ptr == i ? on_value_ : off_value_; **output_ptr = **input_ptr == i ? on_value_ : off_value_;
++output_ptr; ++(*output_ptr);
} }
++input_ptr; ++(*input_ptr);
} else { } else {
const T *in = input_ptr; const T *in = *input_ptr;
for (index_t i = 0; i < length; ++i) { for (index_t i = 0; i < length; ++i) {
input_ptr = in; *input_ptr = in;
run(input, input_ptr, output_ptr, axis, current_in, run(input, input_ptr, output_ptr, axis, current_in,
current_out + 1, left - 1, i); current_out + 1, left - 1, i);
} }
...@@ -135,9 +134,9 @@ class OneHotOp<DeviceType::CPU, T> : public OneHotOpBase { ...@@ -135,9 +134,9 @@ class OneHotOp<DeviceType::CPU, T> : public OneHotOpBase {
if (left == 0) { if (left == 0) {
for (index_t i = 0; i < length; ++i) { for (index_t i = 0; i < length; ++i) {
*output_ptr = *input_ptr == test ? on_value_ : off_value_; **output_ptr = **input_ptr == test ? on_value_ : off_value_;
++output_ptr; ++(*output_ptr);
++input_ptr; ++(*input_ptr);
} }
} else { } else {
for (index_t i = 0; i < length; ++i) { for (index_t i = 0; i < length; ++i) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册