Skip to content

documentation

torch_to_nnef.op.custom_extractors.rnn

Initializing search

documentation

Home
📖 Tutorials
📖 Tutorials
Export API
Internals API
Internals API
- torch_to_nnef
  torch_to_nnef
  - torch_to_nnef
  - collect_env
  - compress
  - console
  - dtypes
  - exceptions
  - export
  - inference_target
    inference_target
    
    torch_to_nnef.inference_target
    
    base
    
    khronos
    
    tract
  - llm_tract
    llm_tract
    
    torch_to_nnef.llm_tract
    
    cli
    
    config
    
    exporter
    
    models
    models
    
    torch_to_nnef.llm_tract.models
    
    base
  - log
  - model_wrapper
  - nnef_graph
  - op
    op
    
    torch_to_nnef.op
    
    aten
    aten
    
    torch_to_nnef.op.aten
    
    activation
    
    attn
    
    axes_change
    
    complex
    
    concat
    
    expand
    
    fft
    
    math
    
    matmul
    
    norm
    
    other
    
    pad
    
    pool
    
    qops
    
    reducer
    
    selector
    
    split
    
    tensor_build
    
    unary
    
    custom_extractors
    custom_extractors
    
    torch_to_nnef.op.custom_extractors
    
    base
    
    rnn rnn
    Table of contents
    
    rnn
    
    LSTMExtractor
    
    fragment
    
    helper
    
    quantized
  - tensor
    tensor
    
    torch_to_nnef.tensor
    
    named
    
    offload
    
    opaque
    
    quant
    quant
    
    torch_to_nnef.tensor.quant
    
    base
    
    qtract
    
    updater
    
    utils
  - torch_graph
    torch_graph
    
    torch_to_nnef.torch_graph
    
    ir_data
    
    ir_graph
    
    ir_helpers
    
    ir_module_tracer
    
    ir_naming
    
    ir_op
    
    torch_const
  - utils
Contributing
Contributing
Why NNEF
Backstory
Changelog

Table of contents

rnn
- LSTMExtractor

rnn

torch_to_nnef.op.custom_extractors.rnn

LSTMExtractor

LSTMExtractor()

Bases: _RNNMixin, ModuleInfoExtractor