Skip to content

N
ngraph
Unverified Commit 49349a57 authored by Robert Kimball's avatar Robert Kimball Committed by GitHub
Browse files
Options

Pretty format for opt_kernels (#4492) 

Co-authored-by: 's avatarSang Ik Lee <sang.ik.lee@intel.com>
parent b007440a
Hide whitespace changes
Inline Side-by-side
Showing with 59 additions and 24 deletions
src/ngraph/runtime/opt_kernel/broadcast.hpp
View file @ 49349a57
...@@ -40,7 +40,11 @@ namespace ngraph ...@@ -40,7 +40,11 @@ namespace ngraph
{ {
for (index[1] = 0; index[1] < out_shape[1]; ++index[1]) for (index[1] = 0; index[1] < out_shape[1]; ++index[1])
{ {
out[index[0] * out_strides[0] + index[1]] = in[in_index]; // clang-format off
out[index[0] * out_strides[0] +
index[1]] =
in[in_index];
// clang-format on
} }
} }
} }
...@@ -59,8 +63,12 @@ namespace ngraph ...@@ -59,8 +63,12 @@ namespace ngraph
{ {
for (index[2] = 0; index[2] < out_shape[2]; ++index[2]) for (index[2] = 0; index[2] < out_shape[2]; ++index[2])
{ {
out[index[0] * out_strides[0] + index[1] * out_strides[1] + index[2]] = // clang-format off
in[in_index]; out[index[0] * out_strides[0] +
index[1] * out_strides[1] +
index[2]] =
in[in_index];
// clang-format on
} }
} }
} }
...@@ -81,8 +89,13 @@ namespace ngraph ...@@ -81,8 +89,13 @@ namespace ngraph
{ {
for (index[3] = 0; index[3] < out_shape[3]; ++index[3]) for (index[3] = 0; index[3] < out_shape[3]; ++index[3])
{ {
out[index[0] * out_strides[0] + index[1] * out_strides[1] + // clang-format off
index[2] * out_strides[2] + index[3]] = in[in_index]; out[index[0] * out_strides[0] +
index[1] * out_strides[1] +
index[2] * out_strides[2] +
index[3]] =
in[in_index];
// clang-format on
} }
} }
} }
...@@ -106,9 +119,14 @@ namespace ngraph ...@@ -106,9 +119,14 @@ namespace ngraph
{ {
for (index[4] = 0; index[4] < out_shape[4]; ++index[4]) for (index[4] = 0; index[4] < out_shape[4]; ++index[4])
{ {
out[index[0] * out_strides[0] + index[1] * out_strides[1] + // clang-format off
index[2] * out_strides[2] + index[3] * out_strides[3] + out[index[0] * out_strides[0] +
index[4]] = in[in_index]; index[1] * out_strides[1] +
index[2] * out_strides[2] +
index[3] * out_strides[3] +
index[4]] =
in[in_index];
// clang-format on
} }
} }
} }
...@@ -135,9 +153,15 @@ namespace ngraph ...@@ -135,9 +153,15 @@ namespace ngraph
{ {
for (index[5] = 0; index[5] < out_shape[5]; ++index[5]) for (index[5] = 0; index[5] < out_shape[5]; ++index[5])
{ {
out[index[0] * out_strides[0] + index[1] * out_strides[1] + // clang-format off
index[2] * out_strides[2] + index[3] * out_strides[3] + out[index[0] * out_strides[0] +
index[4] * out_strides[4] + index[5]] = in[in_index]; index[1] * out_strides[1] +
index[2] * out_strides[2] +
index[3] * out_strides[3] +
index[4] * out_strides[4] +
index[5]] =
in[in_index];
// clang-format on
} }
} }
} }
......
src/ngraph/runtime/opt_kernel/reshape.hpp
View file @ 49349a57
...@@ -76,7 +76,10 @@ namespace ngraph ...@@ -76,7 +76,10 @@ namespace ngraph
{ {
for (in_index[1] = 0; in_index[1] < size[1]; ++in_index[1]) for (in_index[1] = 0; in_index[1] < size[1]; ++in_index[1])
{ {
*out++ = in[*map_index[0] * in_shape[1] + *map_index[1]]; // clang-format off
*out++ = in[*map_index[0] * in_shape[1] +
*map_index[1]];
// clang-format on
} }
} }
} }
...@@ -102,8 +105,11 @@ namespace ngraph ...@@ -102,8 +105,11 @@ namespace ngraph
{ {
for (in_index[2] = 0; in_index[2] < size[2]; ++in_index[2]) for (in_index[2] = 0; in_index[2] < size[2]; ++in_index[2])
{ {
// clang-format off
*out++ = in[*map_index[0] * in_shape[1] * in_shape[2] + *out++ = in[*map_index[0] * in_shape[1] * in_shape[2] +
*map_index[1] * in_shape[2] + *map_index[2]]; *map_index[1] * in_shape[2] +
*map_index[2]];
// clang-format on
} }
} }
} }
...@@ -132,10 +138,13 @@ namespace ngraph ...@@ -132,10 +138,13 @@ namespace ngraph
{ {
for (in_index[3] = 0; in_index[3] < size[3]; ++in_index[3]) for (in_index[3] = 0; in_index[3] < size[3]; ++in_index[3])
{ {
// clang-format off
*out++ = *out++ =
in[*map_index[0] * in_shape[1] * in_shape[2] * in_shape[3] + in[*map_index[0] * in_shape[1] * in_shape[2] * in_shape[3] +
*map_index[1] * in_shape[2] * in_shape[3] + *map_index[1] * in_shape[2] * in_shape[3] +
*map_index[2] * in_shape[3] + *map_index[3]]; *map_index[2] * in_shape[3] +
*map_index[3]];
// clang-format on
} }
} }
} }
...@@ -167,12 +176,14 @@ namespace ngraph ...@@ -167,12 +176,14 @@ namespace ngraph
{ {
for (in_index[4] = 0; in_index[4] < size[4]; ++in_index[4]) for (in_index[4] = 0; in_index[4] < size[4]; ++in_index[4])
{ {
// clang-format off
*out++ = *out++ =
in[*map_index[0] * in_shape[1] * in_shape[2] * in_shape[3] * in[*map_index[0] * in_shape[1] * in_shape[2] * in_shape[3] * in_shape[4] +
in_shape[4] +
*map_index[1] * in_shape[2] * in_shape[3] * in_shape[4] + *map_index[1] * in_shape[2] * in_shape[3] * in_shape[4] +
*map_index[2] * in_shape[3] * in_shape[4] + *map_index[2] * in_shape[3] * in_shape[4] +
*map_index[3] * in_shape[4] + *map_index[4]]; *map_index[3] * in_shape[4] +
*map_index[4]];
// clang-format on
} }
} }
} }
...@@ -207,14 +218,14 @@ namespace ngraph ...@@ -207,14 +218,14 @@ namespace ngraph
{ {
for (in_index[5] = 0; in_index[5] < size[5]; ++in_index[5]) for (in_index[5] = 0; in_index[5] < size[5]; ++in_index[5])
{ {
*out++ = in[*map_index[0] * in_shape[1] * in_shape[2] * // clang-format off
in_shape[3] * in_shape[4] * in_shape[5] + *out++ = in[*map_index[0] * in_shape[1] * in_shape[2] * in_shape[3] * in_shape[4] * in_shape[5] +
*map_index[1] * in_shape[2] * in_shape[3] * *map_index[1] * in_shape[2] * in_shape[3] * in_shape[4] * in_shape[5] +
in_shape[4] * in_shape[5] + *map_index[2] * in_shape[3] * in_shape[4] * in_shape[5] +
*map_index[2] * in_shape[3] * in_shape[4] *
in_shape[5] +
*map_index[3] * in_shape[4] * in_shape[5] + *map_index[3] * in_shape[4] * in_shape[5] +
*map_index[4] * in_shape[5] + *map_index[5]]; *map_index[4] * in_shape[5] +
*map_index[5]];
// clang-format on
} }
} }
} }
......
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