Commit 68a2443a authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Move long return type as tailing return type

parent 10b99e51
...@@ -5,8 +5,6 @@ ...@@ -5,8 +5,6 @@
bool run_permute_bundle(const Problem& problem) bool run_permute_bundle(const Problem& problem)
{ {
constexpr std::size_t NumElemsInBundle = sizeof(BundleType) / sizeof(DataType);
const auto& input_bundle_shape = problem.shape; const auto& input_bundle_shape = problem.shape;
const auto& input_bundle_axes = problem.axes; const auto& input_bundle_axes = problem.axes;
...@@ -49,6 +47,8 @@ bool run_permute_bundle(const Problem& problem) ...@@ -49,6 +47,8 @@ bool run_permute_bundle(const Problem& problem)
output_device_buf.FromDevice(data(output_bundle_tensor)); output_device_buf.FromDevice(data(output_bundle_tensor));
constexpr std::size_t NumElemsInBundle = sizeof(BundleType) / sizeof(DataType);
// extend tensor shape from [N, H, W] to [N, H, W, NumElemsInBundle] // extend tensor shape from [N, H, W] to [N, H, W, NumElemsInBundle]
// axes from [0, 2, 1] to [0, 2, 1, 3] // axes from [0, 2, 1] to [0, 2, 1, 3]
const auto input_shape = extend_shape(input_bundle_shape, NumElemsInBundle); const auto input_shape = extend_shape(input_bundle_shape, NumElemsInBundle);
......
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