diff options
Diffstat (limited to 'utils/GraphUtils.cpp')
-rw-r--r-- | utils/GraphUtils.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/utils/GraphUtils.cpp b/utils/GraphUtils.cpp index 0ce6076f6e..46b748bce6 100644 --- a/utils/GraphUtils.cpp +++ b/utils/GraphUtils.cpp @@ -85,9 +85,11 @@ void CaffePreproccessor::preprocess(ITensor &tensor) Window window; window.use_tensor_dimensions(tensor.info()->tensor_shape()); + const int channel_idx = get_data_layout_dimension_index(tensor.info()->data_layout(), DataLayoutDimension::CHANNEL); + execute_window_loop(window, [&](const Coordinates & id) { - const float value = *reinterpret_cast<float *>(tensor.ptr_to_element(id)) - _mean[id.z()]; + const float value = *reinterpret_cast<float *>(tensor.ptr_to_element(id)) - _mean[id[channel_idx]]; *reinterpret_cast<float *>(tensor.ptr_to_element(id)) = value; }); } |