MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / PermutePrimitiveExists

Function PermutePrimitiveExists

oneflow/user/kernels/transpose_kernel.cpp:93–97  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

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

Callers 1

Calls 1

NewPermutePrimitiveFunction · 0.70

Tested by

no test coverage detected