| 504 | } |
| 505 | |
| 506 | std::vector<unsigned int> BatchMatMulDescriptor::GetAxesNotMul( |
| 507 | DataLayout dataLayout, |
| 508 | const TensorShape& tensorShape) |
| 509 | { |
| 510 | auto axesToMul = BatchMatMulDescriptor::GetAxesToMul(dataLayout, tensorShape); |
| 511 | std::vector<unsigned int> axesNotMul; |
| 512 | for(unsigned int i = 0; i < tensorShape.GetNumDimensions(); i++) |
| 513 | { |
| 514 | if(i == axesToMul.first || i == axesToMul.second) |
| 515 | { |
| 516 | continue; |
| 517 | } |
| 518 | axesNotMul.push_back(i); |
| 519 | } |
| 520 | return axesNotMul; |
| 521 | } |
| 522 | |
| 523 | PermutationVector BatchMatMulDescriptor::GetPermuteVec( |
| 524 | DataLayout dataLayout, |
nothing calls this directly
no test coverage detected