LSTM

Inputs:

  • input0: float32(NC4HW4)
  • input1: float32(NC4HW4, optional)

Outputs:

  • output: float32

Backend:

    • CPU
    • Metal
    • Vulkan: support one input only
    • OpenCL

Caffe op:

  • LSTM