mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-17 21:10:43 +00:00
* GridSample OP implementation for CPU and CUDA **Description**: This change contains implementation for torch grid_sample OP. Cuda implementation contains contribution from Muscle Wu. * Use interpolation for out-of-bound points in zero padding mode Out-of-bound points in zeros padding mode changed from constant 0 to interpolation of surrounding pixels. This aligns with Pytorch implementation. A bug in CUDA batch offset calculation is fixed. Custom op exporter type is added. * Fix nearest bug in CPU * Update per CI build finding and review comments * Force float to avoid potential integer T issue * Style update * PR update * Remove c++17 feature from cuda code |
||
|---|---|---|
| .. | ||
| util | ||
| __init__.py | ||
| convert_onnx_models_to_ort.py | ||
| create_reduced_build_config.py | ||
| dump_ort_model.py | ||
| dump_subgraphs.py | ||
| example_operator_perf_test.py | ||
| FindOptimizerOpsetVersionUpdatesRequired.py | ||
| gen_contrib_doc.py | ||
| gen_opkernel_doc.py | ||
| gen_ort_mobile_pkg_doc.py | ||
| get_submodules.py | ||
| onnx_test_data_utils.py | ||
| ort_test_dir_utils.py | ||
| PythonTools.md | ||
| register_custom_ops_pytorch_exporter.py | ||
| remove_initializer_from_input.py | ||
| run_android_emulator.py | ||
| sparsify_initializers.py | ||
| update_version.py | ||