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
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
float16.h
framework_common.h
framework_provider_common.h
func_api.h
kernel_def_builder.h
kernel_registry.h
op_kernel.h
op_kernel_context.h
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
ortdevice.h
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
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