File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed
Samples/CustomOperator/desktop/cpp/operators Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -24,18 +24,21 @@ struct CustomOperatorProvider :
2424
2525 void RegisterSchemas ()
2626 {
27+ // Register a Relu operator schemain a custom domain ("MyReluDomain" version 1)
2728 ReluOperator ::RegisterReluSchema (m_registry );
29+
30+ // Register a NoisyRelu operator schemain a custom domain ("MyNoisyReluDomain" version 1)
2831 NoisyReluOperatorFactory ::RegisterNoisyReluSchema (m_registry );
2932
3033 DebugOperatorFactory ::RegisterDebugSchema (m_registry );
3134 }
3235
3336 void RegisterKernels ()
3437 {
35- // Replace the Relu operator kernel
38+ // Register a CPU Relu kernel in a custom domain ("MyReluDomain" version 1)
3639 CpuReluOperatorFactory ::RegisterReluKernel (m_registry );
3740
38- // Replace the Relu operator kernel
41+ // Register a GPU Relu kernel in a custom domain ("MyReluDomain" version 1)
3942 GpuReluOperatorFactory ::RegisterReluKernel (m_registry );
4043
4144 // Add a new operator kernel for Relu
You can’t perform that action at this time.
0 commit comments