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

Add input transposing for MatMul (#1462)

Browse Source
This commit is contained in:
Liubov Batanina
2020-07-24 15:15:20 +03:00
committed by GitHub
parent c10c80bdca
commit a38e9f9059
5 changed files with 20 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
inference-engine/tests/ngraph_functions/include/ngraph_functions/builders.hpp
View File

@@ -258,7 +258,9 @@ std::shared_ptr<Node> makeShuffleChannels(const ngraph::Output<Node> &in,
int group);
std::shared_ptr<Node> makeMatMul(const Output<Node> &A,
const Output<Node> &B);
const Output<Node> &B,
bool transpose_a = false,
bool transpose_b = false);
std::shared_ptr<ngraph::Node> makeReduce(std::vector<ngraph::Output<Node>> &in,
const std::vector<int> &reductionAxes,

6
inference-engine/tests/ngraph_functions/src/mat_mul.cpp
View File

@@ -8,8 +8,10 @@ namespace ngraph {
namespace builder {
std::shared_ptr<Node> makeMatMul(const Output<Node>& A,
const Output<Node>& B) {
return std::make_shared<ngraph::opset3::MatMul>(A, B);
const Output<Node>& B,
bool transpose_a,
bool transpose_b) {
return std::make_shared<ngraph::opset3::MatMul>(A, B, transpose_a, transpose_b);
}
} // namespace builder