• Amy Zhuang's avatar
    Create mkldnn primitives at first iteration for codegen - part2 (#2859) · 9335e41c
    Amy Zhuang authored
    * Create mkldnn primitives at first iteration for CODEGEN.
    
     OPs: add, lstm, and rnn.
    
    *  OPs: batchnorm.
    
    *  OPs: concat and lrn.
    
    Remove dead code.
    
    * Skip in place concat, relu, reshape, and slice when building node_primitive_string_deps_index map.
    
    * Change NGRAPH_ASSERT to NGRAPH_CHECK.
    
    * Address PR Feedback.
    
    * Create mkldnn primitives at first iteration for CODEGEN.
     OPs: convertlayout, relu, leakyrelu, boundedrelu, sigmoid, softmax, slice.
    
    * Fix bugs.
    
    *  OPs: quantizedconcat.
    
    Check if there are descriptors before emitting code to read desc_file.
    
    *  OPs: convolution backward.
    
    Use macro to write mkldnn memory dims to generated file.
    
    *  OPs: MaxPoolWithIndices and MaxPoolWithIndicesBackprop.
    
    Add unit tests for MaxPoolWithIndices, MaxPoolWithIndicesBackprop, and MaxPoolBackprop.
    
    * Fix style error.
    
    *  OPs: AvgPoolBackprop and MaxPoolBackprop.
    
    Add unit test for AvgPoolBackprop.
    
    *  OPs: DeconvolutionBias.
    
    *  OPs: Quantize and Dequantize.
    
    *  OPs: QuantizedDot and QuantizedDotBias.
    
    * Use reference kernel for QuantizedConvolution for CODEGEN when mkldnn does not support the parameter types.
    Get scales for quantization ops in cpu_emitter.
    
    * Fix Windows build error: add CPU_BACKEND_API.
    
    * Use template for quantization ops.
    
    *  OPs: QuantizedMatmul.
    
    Emit referece kernel for QuantizedDot in CODEGEN.
    
    * Remove QuantizedDot from get_scale_index.
    
    * Address PR feedback.
    9335e41c
Name
Last commit
Last update
..
files Loading commit data...
models Loading commit data...
onnx Loading commit data...
ref_generators 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...
autodiff.in.cpp Loading commit data...
backend_all.in.cpp Loading commit data...
backend_any.in.cpp Loading commit data...
backend_api.cpp Loading commit data...
backend_api.in.cpp Loading commit data...
backend_arg_reduce.in.cpp Loading commit data...
backend_binary_elementwise.in.cpp Loading commit data...
backend_broadcast.in.cpp Loading commit data...
backend_comparison.in.cpp Loading commit data...
backend_debug_api.cpp Loading commit data...
backend_dot.in.cpp Loading commit data...
backend_embedding_lookup.in.cpp Loading commit data...
backend_fused_op.in.cpp Loading commit data...
backend_gather.in.cpp Loading commit data...
backend_graph_comparison.in.cpp Loading commit data...
backend_one_hot.in.cpp Loading commit data...
backend_performance.cpp Loading commit data...
backend_pool.in.cpp Loading commit data...
backend_reshape.in.cpp Loading commit data...
backend_scatter.in.cpp Loading commit data...
backend_sum.in.cpp Loading commit data...
backend_test.in.cpp Loading commit data...
backend_topk.in.cpp Loading commit data...
backend_unary_elementwise.in.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...
convolution_test.in.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_debugger.cpp Loading commit data...
cpu_fusion.cpp Loading commit data...
cpu_test.cpp Loading commit data...
cse.cpp Loading commit data...
distributed.in.cpp Loading commit data...
dynamic.in.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...
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_shapes.cpp Loading commit data...
tensor.cpp Loading commit data...
tools.cpp Loading commit data...
type_prop.cpp Loading commit data...
type_prop_layers.cpp Loading commit data...
update_reference.sh Loading commit data...
util.cpp Loading commit data...
zero_dim_tensor_elimination.cpp Loading commit data...