• Pruthvi's avatar
    LSTM MKLDNN integration for ONNX LSTM op (#3327) · e5d606b8
    Pruthvi authored
    * - Add graph pass method for onnx lstmcell rewrite with lstm cpu op
    - insert reshapes to keep the weights in ldigo format
    - test case for onnx LstmCell to CPU Lstm
    
    * fix typo
    
    * - check LSTMCell for the fused op decomposistion in the backend
    
    * - fix bug in onnx_lstm graph pass
    - passes unit test
    
    * style-fix
    
    * - fix compilation error
    - use IFCO gate ordering for bias
    
    *  - Skip LSTMCell to LSTM CPU fusion for peephole
    
    * - add comment && remove duplicate function
    
    * -use dynamic_pointer_cast to check for constant
    
    * - onnx bias will be of shape (2 * gates_count * hidden_size) bias of Wb and Rb are concatenated, we will split the bias, add and rearrange in order IFCO
    
    * - Use most derived LSTM ctor for pattern matching
    
    * - Style Fix
    
    * style fix
    
    * Address PR comments
    
    * - add support for graph pass (MKLDNN version > 1) for mapping LSTMCell -> LSTM CPU op
    
    * fix unit test failure for MKLDNN V1.0
    e5d606b8
Name
Last commit
Last update
..
backend Loading commit data...
files Loading commit data...
models Loading commit data...
onnx Loading commit data...
ref_generators Loading commit data...
type_prop Loading commit data...
util Loading commit data...
CMakeLists.txt Loading commit data...
algebraic_simplification.cpp Loading commit data...
aligned_buffer.cpp Loading commit data...
all_close_f.cpp Loading commit data...
assertion.cpp Loading commit data...
backend_api.cpp Loading commit data...
backend_debug_api.cpp Loading commit data...
backend_performance.cpp Loading commit data...
bfloat16.cpp Loading commit data...
build_graph.cpp Loading commit data...
builder.cpp Loading commit data...
builder_autobroadcast.cpp Loading commit data...
builder_quantization.cpp Loading commit data...
check.cpp Loading commit data...
concat_fusion.cpp Loading commit data...
constant_folding.cpp Loading commit data...
control_dependencies.cpp Loading commit data...
coordinate.cpp Loading commit data...
copy.cpp Loading commit data...
core.cpp Loading commit data...
core_fusion.cpp Loading commit data...
cpio.cpp Loading commit data...
cpu_codegen.cpp Loading commit data...
cpu_debug_tracer.cpp Loading commit data...
cpu_debugger.cpp Loading commit data...
cpu_fusion.cpp Loading commit data...
cpu_test.cpp Loading commit data...
cse.cpp Loading commit data...
dyn_elimination.cpp Loading commit data...
element_type.cpp Loading commit data...
event_tracing.cpp Loading commit data...
file_util.cpp Loading commit data...
float16.cpp Loading commit data...
gpu_fusion.cpp Loading commit data...
gpu_test.cpp Loading commit data...
halide.cpp Loading commit data...
hybrid_backend.cpp Loading commit data...
includes.cpp Loading commit data...
input_output_assign.cpp Loading commit data...
main.cpp Loading commit data...
misc.cpp Loading commit data...
misc.hpp Loading commit data...
mkldnn.cpp Loading commit data...
ngraph_api.cpp Loading commit data...
node_input_output.cpp Loading commit data...
nop_elimination.cpp Loading commit data...
op.cpp Loading commit data...
partial_shape.cpp Loading commit data...
pass.cpp Loading commit data...
pass_liveness.cpp Loading commit data...
pass_manager.cpp Loading commit data...
pass_memory_layout.cpp Loading commit data...
pass_shape_relevance.cpp Loading commit data...
pattern.cpp Loading commit data...
provenance.cpp Loading commit data...
reshape_elimination.cpp Loading commit data...
reshape_sinking.cpp Loading commit data...
serialize.cpp Loading commit data...
shape.cpp Loading commit data...
specialize_function.cpp Loading commit data...
tensor.cpp Loading commit data...
tools.cpp Loading commit data...
type_prop_benchmark.cpp Loading commit data...
type_prop_layers.cpp Loading commit data...
update_convolution_reference.sh Loading commit data...
update_dyn_replace_slice_reference.sh Loading commit data...
update_dyn_slice_reference.sh Loading commit data...
util.cpp Loading commit data...
zero_dim_tensor_elimination.cpp Loading commit data...