Uploaded image for project: 'Apache MXNet (Retired)'
  1. Apache MXNet (Retired)
  2. MXNET-107

Fused RNN operator for CPU

    XMLWordPrintableJSON

Details

    • New Feature
    • Status: To Do
    • Major
    • Resolution: Unresolved
    • None
    • None

    Description

      Since fused RNN operator for GPU is already implemented with cudnn, I think we should also provide the same functionality at CPU side. So user models which call sym.RNN can run both on GPU and CPU. This will also improve the performance of RNN related models a lot on CPU.

      Attachments

        Activity

          People

            Unassigned Unassigned
            taolv Tao Lv
            Votes:
            0 Vote for this issue
            Watchers:
            1 Start watching this issue

            Dates

              Created:
              Updated:

              Time Tracking

                Estimated:
                Original Estimate - Not Specified
                Not Specified
                Remaining:
                Remaining Estimate - 0h
                0h
                Logged:
                Time Spent - 17h
                17h