custom-op icon indicating copy to clipboard operation
custom-op copied to clipboard

how to make a fastertransformer custom op

Open 652994331 opened this issue 5 years ago • 0 comments

Hi, i already used your zero out custom op successfully , now i am trying to build a fastertransformer custom op accroding to this.https://github.com/NVIDIA/DeepLearningExamples/tree/master/FasterTransformer/v1/fastertransformer/tf_op. However, i feel it's pretty hard to modify the zero_out custom op BUILD file to a fastertransformer. it has a lot of cc_library, py_library which i am not sure how to modify it. I was wondering, can u give me some advices on how to modify it, thanks

652994331 avatar Oct 08 '20 19:10 652994331