| 36 | std::vector<op_desc> operators() const { return {{"Transpose"}}; } |
| 37 | |
| 38 | instruction_ref parse(const op_desc& /*opd*/, |
| 39 | const tf_parser& /*parser*/, |
| 40 | const tf_parser::node_info& info, |
| 41 | std::vector<instruction_ref> args) const |
| 42 | { |
| 43 | auto perm = args[1]->eval().get<int32_t>().to_vector(); |
| 44 | std::vector<int64_t> dims(perm.begin(), perm.end()); |
| 45 | |
| 46 | return info.add_instruction(make_op("transpose", {{"permutation", dims}}), args.front()); |
| 47 | } |
| 48 | }; |
| 49 | |
| 50 | } // namespace tf |