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

[IE CLDNN] Convert permutation order in MatMul (#6005)

Browse Source
This commit is contained in:
Ewa Tusień 2021-06-07 11:58:22 +02:00 committed by GitHub
parent e2ffcae852
commit a30d757cfd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
inference-engine/src/cldnn_engine/ops/matmul.cpp
View File

@ -83,10 +83,11 @@ void CreateMatMulOp(Program& p, const std::shared_ptr<ngraph::op::v0::MatMul>& o
for (auto o = transpose_order.size(); o < 4; o++) for (auto o = transpose_order.size(); o < 4; o++)
transpose_order.push_back((uint16_t)o); transpose_order.push_back((uint16_t)o);
std::vector<uint16_t> cldnn_permute_order = ConvertPermuteOrder(transpose_order);
auto permuteName = op->get_friendly_name() + "/transpose_b"; auto permuteName = op->get_friendly_name() + "/transpose_b";
auto permutePrim = cldnn::permute(permuteName, auto permutePrim = cldnn::permute(permuteName,
weightsName, weightsName,
transpose_order); cldnn_permute_order);
p.AddPrimitive(permutePrim); p.AddPrimitive(permutePrim);
p.AddInnerPrimitiveToProfiler(permuteName, layerName, op); p.AddInnerPrimitiveToProfiler(permuteName, layerName, op);
weightsName = permuteName; weightsName = permuteName;
@ -102,10 +103,11 @@ void CreateMatMulOp(Program& p, const std::shared_ptr<ngraph::op::v0::MatMul>& o
for (auto o = transpose_order.size(); o < 4; o++) for (auto o = transpose_order.size(); o < 4; o++)
transpose_order.push_back((uint16_t)o); transpose_order.push_back((uint16_t)o);
std::vector<uint16_t> cldnn_permute_order = ConvertPermuteOrder(transpose_order);
auto permuteName = op->get_friendly_name() + "/transpose_a"; auto permuteName = op->get_friendly_name() + "/transpose_a";
auto permutePrim = cldnn::permute(permuteName, auto permutePrim = cldnn::permute(permuteName,
inputName, inputName,
transpose_order); cldnn_permute_order);
p.AddPrimitive(permutePrim); p.AddPrimitive(permutePrim);
p.AddInnerPrimitiveToProfiler(permuteName, layerName, op); p.AddInnerPrimitiveToProfiler(permuteName, layerName, op);
inputName = permuteName; inputName = permuteName;