| 91 | }; |
| 92 | |
| 93 | auto PermutePrimitiveExists() { |
| 94 | return hob::make_custom("PermutePrimitiveExists", [](const user_op::KernelRegContext& ctx) { |
| 95 | return NewPermutePrimitive(&ctx).operator bool(); |
| 96 | }); |
| 97 | } |
| 98 | |
| 99 | REGISTER_USER_KERNEL("transpose") |
| 100 | .SetCreateFn<TransposeKernel>() |
no test coverage detected