saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-20 21:40:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/include/onnxruntime/core/framework
History
Vincent Wang 5ecfaef042
ATen Fallback for Inference (#11597) 
* aten op for inference

* fix build error

* more some code to training only

* remove domain from operator name

* move aten_op_executor ext out from ortmodule

* add pipeline

* add exec mode

* fix script

* fix ut script

* fix test pipeline

* failure test

* rollback

* bugfix

* resolve comments

* enable aten for python build only

* fix win build

* use target_compile_definitions

* support io binding

* turn off aten by default

* fix ut

Co-authored-by: Vincent Wang <weicwang@microsoft.com>
Co-authored-by: zhijxu <zhijxu@microsoft.com>
2022-06-09 16:07:30 +08:00
..
alloc_kind.h Add New AllocKind for YieldOp Outputs, Run YieldOp with InferenceSession in UT (#7125) 2021-03-25 15:18:51 +08:00
allocator.h support MIGraphXEP to work with ROCMEP for inference on AMD GPU (#10368) 2022-01-26 15:52:56 -08:00
buffer_deleter.h Add ability for pre-packed weights of shared initializers to be shared across sessions (#7421) 2021-05-14 20:44:42 -07:00
customregistry.h Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
data_types.h Consolidate utils::ToTensorProtoElementType, TypeToDataType, and data_types_internal::ToTensorDataType. (#9824) 2022-04-20 12:45:53 -07:00
data_types_internal.h Consolidate utils::ToTensorProtoElementType, TypeToDataType, and data_types_internal::ToTensorDataType. (#9824) 2022-04-20 12:45:53 -07:00
endian.h
execution_provider.h Rework allocator sharing to work for multiple devices. (#11700) 2022-06-09 17:38:38 +10:00
fence.h
float16.h fix build error (#11199) 2022-04-13 13:09:19 +08:00
framework_common.h
func_api.h Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
kernel_def_builder.h enable strided tensor for training only (#10748) 2022-03-08 08:31:28 +08:00
kernel_registry.h APIs for custom op to invoke ort operator directly (#10713) 2022-05-03 14:16:30 -07:00
op_kernel.h Snpe ep (#11665) 2022-06-03 14:10:02 -07:00
op_kernel_context.h ATen Fallback for Inference (#11597) 2022-06-09 16:07:30 +08:00
op_kernel_info.h Fix SDL warnings in CPU EP (#9975) 2021-12-19 20:54:29 -08:00
op_node_proto_helper.h Reduce number of memory allocations based on a customer profiling case (#10193) 2022-01-24 10:40:46 -08:00
ort_value.h Support disabling support for the optional type in ORT builds (#9745) 2021-11-17 19:13:28 -08:00
ortdevice.h Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
ortmemoryinfo.h C API Enum Name Fixes (#9092) 2021-09-17 15:11:26 -07:00
provider_options.h Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
provider_options_utils.h Plug n Allocate with external CUDA allocator via PyBind. (#6679) 2021-02-17 18:59:38 -08:00
provider_shutdown.h
run_options.h Add ability for memory arenas to "shrink" periodically (#7284) 2021-05-08 07:53:21 -07:00
sparse_tensor.h Make SparseTensor infrastructure optional (#8802) 2021-08-27 17:12:26 +10:00
tensor.h Create Tensor with Strides (#11294) 2022-04-28 16:49:37 +08:00
tensor_shape.h Introduce a way to disable Abseil library (#11353) 2022-04-27 08:57:52 -07:00
to_tensor_proto_element_type.h Consolidate utils::ToTensorProtoElementType, TypeToDataType, and data_types_internal::ToTensorDataType. (#9824) 2022-04-20 12:45:53 -07:00