onnxruntime/include/onnxruntime/core/framework
cao lei 2c525a79b1
Add new API KernelContext_GetScratchBuffer (#19809)
### Description
<!-- Describe your changes. -->
add new API KernelContext_GetScratchBuffer to get scratch buffer from
kernel context


### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
add new API KernelContext_GetScratchBuffer to get scratch buffer from
kernel context which will be used in ORT extension project for
GroupQueryAttention custom op
2024-03-13 19:41:15 -07:00
..
alloc_kind.h
allocator.h Throw if unique_ptr or array allocation fails due to SafeInt overflow (#18941) 2024-01-03 07:57:51 +10:00
buffer_deleter.h
customregistry.h
data_types.h [wasm] upgrade emsdk to 3.1.44 (#17069) 2023-08-10 16:08:36 -07:00
data_types_internal.h Implement CUDA IsInf-10,20 (#19772) 2024-03-05 13:33:01 -08:00
endian.h
execution_provider.h cuda graph enhancement (#19636) 2024-03-07 10:15:18 -08:00
float8.h implement isinf20 and isnan20 (#17874) 2023-10-24 10:58:54 -07:00
float16.h [C#, CPP] Introduce Float16/BFloat16 support and tests for C#, C++ (#16506) 2023-07-14 10:46:52 -07:00
framework_common.h
framework_provider_common.h
func_api.h
kernel_def_builder.h Remove onnxruntime_PYBIND_EXPORT_OPSCHEMA definition from onnxruntime (#15776) 2023-05-03 13:08:35 -07:00
kernel_registry.h Remove onnxruntime_PYBIND_EXPORT_OPSCHEMA definition from onnxruntime (#15776) 2023-05-03 13:08:35 -07:00
op_kernel.h [wasm] upgrade emsdk to 3.1.44 (#17069) 2023-08-10 16:08:36 -07:00
op_kernel_context.h ExecutionProvider API refactor - replace OrtMemoryInfo with OrtDevice (#15618) 2023-05-01 10:06:00 -07:00
op_kernel_info.h Make session configuration options available to kernels via OpKernelInfo (#18897) 2024-01-13 10:02:43 +10:00
op_node_proto_helper.h MLAS AArch64 quantized int4 Gemm kernel (#18031) 2023-11-15 09:31:54 -08:00
ort_value.h Two fixes involving minimal builds (#17000) 2023-08-23 16:01:22 +10:00
ortdevice.h ExecutionProvider API refactor - replace OrtMemoryInfo with OrtDevice (#15618) 2023-05-01 10:06:00 -07:00
ortmemoryinfo.h
provider_options.h
provider_options_utils.h
provider_shutdown.h
run_options.h
sparse_tensor.h
stream_handles.h Add new API KernelContext_GetScratchBuffer (#19809) 2024-03-13 19:41:15 -07:00
tensor.h Handle string initializers in constant folding (#17422) 2023-09-27 21:15:58 +10:00
tensor_shape.h Make TensorShapeVector to use InlinedVector<Int64_t> to reduce on template instantiations (#18519) 2023-11-21 14:13:50 -08:00
to_tensor_proto_element_type.h [wasm] upgrade emsdk to 3.1.44 (#17069) 2023-08-10 16:08:36 -07:00