saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-27 22:45:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/include/onnxruntime/core/framework
History
Hector Li ffb4b62826
Fix allocator issue for TensorRT IOBinding (#6240) 
* Fix issue: https://github.com/microsoft/onnxruntime/issues/6094

Root cause: we didn't expose the OrtMemoryInfo for TRT, so it will cause issue if user want use IObinding for Tensorrt.

Short term fix, add the OrtMemoryInfo for TRT. Long term should unify the allocator for CUDA and TRT
2020-12-31 20:15:43 -08:00
..
alloc_kind.h Exclude some training specific code from the minimal build. Cleanup some related aspects of allocation planner. (#5861) 2020-11-20 20:25:46 +10:00
allocator.h Fix allocator issue for TensorRT IOBinding (#6240) 2020-12-31 20:15:43 -08:00
customregistry.h
data_types.h Enable varied input type for custom op (#6066) 2020-12-09 15:10:42 -08:00
data_types_internal.h
endian.h
execution_provider.h Helper for compiling EP to generate deterministic unique ids for use in MetaDef names (#6156) 2020-12-21 12:17:58 +10:00
fence.h Remove Cuda dependency from TensorRT shared provider (#5014) 2020-09-04 11:35:02 -07:00
framework_common.h
func_api.h Unify IExecutionProvider and IExecutionProviderFactory interfaces (#6108) 2020-12-15 16:45:53 -08:00
kernel_def_builder.h Introduce VariadicAlias, remove hardcoded alias limits (#6106) 2020-12-11 10:47:08 -08:00
kernel_registry.h Add kernel def hash logic for minimal build (#4891) 2020-08-23 14:39:07 +10:00
ml_value.h Support for Sparse Initializers (#5540) 2020-10-27 10:32:06 -07:00
op_kernel.h Unify activation and initializer alignment value (#6109) 2020-12-14 13:13:41 -08:00
op_kernel_info.h Support EPs that compile nodes in a minimal build. (#5776) 2020-11-17 13:52:22 +10:00
op_node_proto_helper.h Allow Kernels refer to some attribute data directly in the protobuf (#5624) 2020-10-29 16:12:54 -07:00
ortdevice.h Deprecate Python global configuration functions [Part 1] (#5923) 2020-12-15 11:32:43 -08:00
ortmemoryinfo.h Remove Cuda dependency from TensorRT shared provider (#5014) 2020-09-04 11:35:02 -07:00
provider_options.h Deprecate Python global configuration functions [Part 1] (#5923) 2020-12-15 11:32:43 -08:00
provider_options_utils.h Deprecate Python global configuration functions [Part 1] (#5923) 2020-12-15 11:32:43 -08:00
provider_shutdown.h Fix shared provider unload crash (#5553) 2020-10-21 13:01:21 -07:00
run_options.h
sparse_tensor.h Support for Sparse Initializers (#5540) 2020-10-27 10:32:06 -07:00
tensor.h Support EPs that compile nodes in a minimal build. (#5776) 2020-11-17 13:52:22 +10:00
tensor_shape.h