• Tomasz Dołbniak's avatar
    RoiAlign operator - ONNX importer + core (#4432) · e1ebbf12
    Tomasz Dołbniak authored
    * ROIAlign op skeleton
    
    * Expose ROIAlign via opset2
    
    * ROIAlign attribute visitor
    
    * Batch indices input for ROIAlign
    
    * Shape inference for ROIAlign
    
    * ROIAlign mode attribute validation
    
    * onnx_importer ROIAlign op
    
    * size_t of get_attribute_value in onnx_importer::Node
    
    * Explicit cast size_t -> int64_t when creating Dimensions
    
    * ...aaand some new lines to make clang format happy
    
    * Rank check fix for the batch indices input.
    Co-Authored-By: 's avatarAdam Osewski <adam.osewski@intel.com>
    
    * Typo
    Co-Authored-By: 's avatarAdam Osewski <adam.osewski@intel.com>
    
    * Empty opset3 definition
    
    * Move ROIAlign to opse3
    
    * Review comments
    
    * Correct rank check for batch indices input
    
    * Basic shape inference test for ROIAlign
    
    * Move ROIAlign to ops from experimental/layers
    
    * ROIAlign -> RoiAlign
    Co-Authored-By: 's avatarKatarzyna Mitrus <katarzyna.mitrus@intel.com>
    
    * Move ROIAlign from v0 to v3
    
    * Support more data types in ROIAlign
    
    * Move the ROIAlign from opset2_tbl to opset3_tbl
    
    * Don't include opset2 in opset3.hpp
    
    * PoolingMode enum for ROIAlign attribute
    
    * Use EnumNames to handle string/enum conversion
    
    * More checks to prevent segfaults
    
    * Remove the std prefix for string
    Co-Authored-By: 's avatarAdam Osewski <adam.osewski@intel.com>
    
    * Fix more potential segfaults
    
    * Build break fix
    
    * ROIAlign serializer
    
    * PR feedback
    
    * Disable the ROIAlign op in GPU backend
    
    * ROIAlign type prop UT
    
    * More shape inference UT
    
    * PR comments for RoiAlign
    
    * code formatting
    Co-authored-by: 's avatarAdam Osewski <adam.osewski@intel.com>
    Co-authored-by: 's avatarKatarzyna Mitrus <katarzyna.mitrus@intel.com>
    Co-authored-by: 's avatarRobert Kimball <robert.kimball@intel.com>
    e1ebbf12
Name
Last commit
Last update
..
backend Loading commit data...
files Loading commit data...
mlir Loading commit data...
models Loading commit data...
onnx Loading commit data...
opset_pass 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...
attributes.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.cpp Loading commit data...
constant_folding.cpp Loading commit data...
control_dependencies.cpp Loading commit data...
convert_u1_to_string.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...
file_util.cpp Loading commit data...
float16.cpp Loading commit data...
gpu_fusion.cpp Loading commit data...
gpu_test.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...
op_is.cpp Loading commit data...
opset1.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...
replace_node.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...