| 2661 | } // namespace |
| 2662 | |
| 2663 | std::map<OperatorType, std::unique_ptr<BaseOperator>> BuildOperatorByTypeMap( |
| 2664 | bool enable_select_tf_ops) { |
| 2665 | std::map<OperatorType, std::unique_ptr<BaseOperator>> result; |
| 2666 | |
| 2667 | std::vector<std::unique_ptr<BaseOperator>> ops = |
| 2668 | BuildOperatorList(enable_select_tf_ops); |
| 2669 | for (auto& op : ops) { |
| 2670 | result[op->type()] = std::move(op); |
| 2671 | } |
| 2672 | |
| 2673 | return result; |
| 2674 | } |
| 2675 | |
| 2676 | std::map<string, std::unique_ptr<BaseOperator>> BuildOperatorByNameMap( |
| 2677 | bool enable_select_tf_ops) { |