onnxruntime/onnxruntime/core/session
Chen Fu be08b47e7b
Refine cast optimizer for safety (#15658)
### Description

Cast optimizer may convert a fp16 node to fp32. This used to be safe as
all fp16 kernels has fp32 implementation. As this assumption is no
longer true, we need to check the validity of the operation



### Motivation and Context

Main work here is to introduce an API to check whether a kernel is
registered. Currently we don't have a way to do that without an operator
node. This needs to be augmented. We need to query whether a kernel is
registered by its property only, so that we can judge whether it is safe
to construct a node long before we actually do so.
2023-04-28 09:32:54 -07:00
..
abi_session_options.cc Run clang-format in CI (#15524) 2023-04-18 09:26:58 -07:00
abi_session_options_impl.h Improve custom op library handle cleanup (#14099) 2023-01-04 17:56:29 -08:00
allocator_adapters.cc
allocator_adapters.h
api_utils.cc Custom Op runtime wrapper (#13427) 2023-01-18 09:09:32 -08:00
api_utils.h Custom Op runtime wrapper (#13427) 2023-01-18 09:09:32 -08:00
custom_ops.cc Single-schema-multi-kernel (#15184) 2023-04-27 13:39:59 -07:00
custom_ops.h Changes to support standalone custom ops in a minimal build. (#14497) 2023-03-01 11:22:54 +10:00
default_cpu_allocator_c_api.cc Deprecate CustomApi and refactor public API for better safety and consistency (#13215) 2022-10-06 14:57:37 -07:00
environment.cc Enable a single build with optimized inference and on device training (#14241) 2023-01-12 21:36:43 -08:00
inference_session.cc Refine cast optimizer for safety (#15658) 2023-04-28 09:32:54 -07:00
inference_session.h Add TRT plugins support using custom ops (#13847) 2023-04-18 20:24:32 -07:00
inference_session_utils.cc Automatically enable tunable op usage for production models (#15156) 2023-04-06 13:52:47 +08:00
inference_session_utils.h Offline tuning (#14558) 2023-02-15 14:17:34 +08:00
IOBinding.cc remove device_id parameter out of ExecutionProvider::GetAllocator() (#14580) 2023-02-13 10:01:07 -08:00
IOBinding.h remove device_id parameter out of ExecutionProvider::GetAllocator() (#14580) 2023-02-13 10:01:07 -08:00
onnxruntime_c_api.cc Create a new C API KernelContext_GetAllocator() for Custom Op scenario (#15591) 2023-04-23 21:54:35 -07:00
ort_apis.h Create a new C API KernelContext_GetAllocator() for Custom Op scenario (#15591) 2023-04-23 21:54:35 -07:00
ort_env.cc Run clang-format in CI (#15524) 2023-04-18 09:26:58 -07:00
ort_env.h
provider_bridge_ort.cc Add TRT plugins support using custom ops (#13847) 2023-04-18 20:24:32 -07:00
provider_bridge_ort.h
provider_registration.cc [js/webgpu] following up for JSEP/WebGPU code cleanup (#15666) 2023-04-25 21:20:03 -07:00
standalone_op_invoker.cc Changes to support standalone custom ops in a minimal build. (#14497) 2023-03-01 11:22:54 +10:00