custom-op
custom-op copied to clipboard
how to make a fastertransformer custom op
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