| 1649 | } |
| 1650 | |
| 1651 | std::string GetOpStr(uint32_t op_type) |
| 1652 | { |
| 1653 | op_tm_name_map_t::const_iterator it = gTmOpName.find(op_type); |
| 1654 | if(it != gTmOpName.end()) |
| 1655 | { |
| 1656 | return it->second; |
| 1657 | } |
| 1658 | |
| 1659 | switch(op_type) |
| 1660 | { |
| 1661 | case TM2_OPTYPE_ACCURACY: |
| 1662 | return std::string(TM2_OPSTR_ACCURACY); |
| 1663 | case TM2_OPTYPE_BATCHNORMALIZATION: |
| 1664 | return std::string(TM2_OPSTR_BATCHNORMALIZATION); |
| 1665 | case TM2_OPTYPE_BILINEARRESIZE: |
| 1666 | return std::string(TM2_OPSTR_BILINEARRESIZE); |
| 1667 | case TM2_OPTYPE_CONCAT: |
| 1668 | return std::string(TM2_OPSTR_CONCAT); |
| 1669 | case TM2_OPTYPE_CONST: |
| 1670 | return std::string(TM2_OPSTR_CONST); |
| 1671 | case TM2_OPTYPE_CONVOLUTION: |
| 1672 | return std::string(TM2_OPSTR_CONVOLUTION); |
| 1673 | case TM2_OPTYPE_DECONVOLUTION: |
| 1674 | return std::string(TM2_OPSTR_DECONVOLUTION); |
| 1675 | case TM2_OPTYPE_DETECTIONOUTPUT: |
| 1676 | return std::string(TM2_OPSTR_DETECTIONOUTPUT); |
| 1677 | case TM2_OPTYPE_DROPOUT: |
| 1678 | return std::string(TM2_OPSTR_DROPOUT); |
| 1679 | case TM2_OPTYPE_ELTWISE: |
| 1680 | return std::string(TM2_OPSTR_ELTWISE); |
| 1681 | case TM2_OPTYPE_FLATTEN: |
| 1682 | return std::string(TM2_OPSTR_FLATTEN); |
| 1683 | case TM2_OPTYPE_FULLYCONNECTED: |
| 1684 | return std::string(TM2_OPSTR_FULLYCONNECTED); |
| 1685 | case TM2_OPTYPE_INPUTOP: |
| 1686 | return std::string(TM2_OPSTR_INPUTOP); |
| 1687 | case TM2_OPTYPE_LRN: |
| 1688 | return std::string(TM2_OPSTR_LRN); |
| 1689 | case TM2_OPTYPE_NORMALIZE: |
| 1690 | return std::string(TM2_OPSTR_NORMALIZE); |
| 1691 | case TM2_OPTYPE_PERMUTE: |
| 1692 | return std::string(TM2_OPSTR_PERMUTE); |
| 1693 | case TM2_OPTYPE_POOLING: |
| 1694 | return std::string(TM2_OPSTR_POOLING); |
| 1695 | case TM2_OPTYPE_PRELU: |
| 1696 | return std::string(TM2_OPSTR_PRELU); |
| 1697 | case TM2_OPTYPE_PRIORBOX: |
| 1698 | return std::string(TM2_OPSTR_PRIORBOX); |
| 1699 | case TM2_OPTYPE_REGION: |
| 1700 | return std::string(TM2_OPSTR_REGION); |
| 1701 | case TM2_OPTYPE_RELU: |
| 1702 | return std::string(TM2_OPSTR_RELU); |
| 1703 | case TM2_OPTYPE_RELU6: |
| 1704 | return std::string(TM2_OPSTR_RELU6); |
| 1705 | case TM2_OPTYPE_REORG: |
| 1706 | return std::string(TM2_OPSTR_REORG); |
| 1707 | case TM2_OPTYPE_RESHAPE: |
| 1708 | return std::string(TM2_OPSTR_RESHAPE); |
no test coverage detected