IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixed 'is_rotating_except_batch' to follow the IE order (#12050)

Browse Source
This commit is contained in:
Eddy Kim
2022-07-08 15:36:17 +09:00
committed by GitHub
parent b5c058738d
commit 8d852b4aee
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/plugins/intel_gpu/src/graph/include/permute_inst.h
View File

@@ -26,13 +26,13 @@ public:
std::vector<uint16_t> get_permute_order() const { return get_primitive()->permute_order; }
bool is_rotating_except_batch() const {
// Target transform: Rotate feature dim to back to be taken as inner-most axis
// ex) 0(b), 4(f), 1(z), 2(y), 3(x)
// ex) 0(b), 3(f), 1(y), 2(x)
// ex) 0(b), 2(f), 3(x), 4(y), 1(z)
// ex) 0(b), 2(f), 3(x), 1(y)
auto& order = get_primitive()->permute_order;
if ((int32_t) order[1] != order.size() - 1) return false;
if ((int32_t) order[order.size() - 2] != order.size() - 1) return false;
if ((int32_t) order[0] != 0) return false;
for (int32_t i = 2; i < (int32_t) order.size(); ++i) {
if ((int32_t)order[i] != (i - 1)) return false;
if ((int32_t)order[i - 1] != i) return false;
}
return true;
}