• 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
..
all.cpp Loading commit data...
any.cpp Loading commit data...
avg_pool.cpp Loading commit data...
batch_mat_mul.cpp Loading commit data...
batch_mat_mul_transpose.cpp Loading commit data...
batch_norm.cpp Loading commit data...
batch_to_space.cpp Loading commit data...
binary_elementwise.cpp Loading commit data...
broadcast.cpp Loading commit data...
clamp.cpp Loading commit data...
compat.cpp Loading commit data...
concat.cpp Loading commit data...
constant.cpp Loading commit data...
convert.cpp Loading commit data...
convolution.cpp Loading commit data...
convolution_bias.cpp Loading commit data...
crop_and_resize.cpp Loading commit data...
deformable_psroi_pooling.cpp Loading commit data...
depth_to_space.cpp Loading commit data...
dequantize.cpp Loading commit data...
dot.cpp Loading commit data...
dyn_broadcast.cpp Loading commit data...
dyn_pad.cpp Loading commit data...
dyn_replace_slice.cpp Loading commit data...
dyn_reshape.cpp Loading commit data...
dyn_slice.cpp Loading commit data...
elu.cpp Loading commit data...
embedding_lookup.cpp Loading commit data...
fake_quantize.cpp Loading commit data...
gather.cpp Loading commit data...
gather_nd.cpp Loading commit data...
gather_tree.cpp Loading commit data...
gemm.cpp Loading commit data...
get_output_element.cpp Loading commit data...
grn.cpp Loading commit data...
group_convolution.cpp Loading commit data...
group_convolution_backprop_data.cpp Loading commit data...
gru_cell.cpp Loading commit data...
hard_sigmoid.cpp Loading commit data...
index_reduction.cpp Loading commit data...
layer_norm.cpp Loading commit data...
lrn.cpp Loading commit data...
lstm_cell.cpp Loading commit data...
lstm_sequence.cpp Loading commit data...
matmul.cpp Loading commit data...
max_pool.cpp Loading commit data...
mvn.cpp Loading commit data...
non_max_suppression.cpp Loading commit data...
non_zero.cpp Loading commit data...
normalize.cpp Loading commit data...
one_hot.cpp Loading commit data...
pad.cpp Loading commit data...
parameter.cpp Loading commit data...
prelu.cpp Loading commit data...
proposal.cpp Loading commit data...
quantize.cpp Loading commit data...
quantized_convolution.cpp Loading commit data...
quantized_dot.cpp Loading commit data...
random_uniform.cpp Loading commit data...
range.cpp Loading commit data...
reduce_prod.cpp Loading commit data...
reduce_sum.cpp Loading commit data...
replace_slice.cpp Loading commit data...
reshape.cpp Loading commit data...
reverse.cpp Loading commit data...
reverse_sequence.cpp Loading commit data...
rnn_cell.cpp Loading commit data...
roi_align.cpp Loading commit data...
scale_shift.cpp Loading commit data...
scatter_add.cpp Loading commit data...
scatter_nd.cpp Loading commit data...
select.cpp Loading commit data...
shape_of.cpp Loading commit data...
shuffle_channels.cpp Loading commit data...
slice.cpp Loading commit data...
space_to_batch.cpp Loading commit data...
space_to_depth.cpp Loading commit data...
split.cpp Loading commit data...
squared_difference.cpp Loading commit data...
squeeze.cpp Loading commit data...
strided_slice.cpp Loading commit data...
sum.cpp Loading commit data...
tile.cpp Loading commit data...
top_k.cpp Loading commit data...
transpose.cpp Loading commit data...
unary_elementwise.cpp Loading commit data...
unsqueeze.cpp Loading commit data...
variadic_split.cpp Loading commit data...