onnxruntime/onnxruntime/test/framework
Scott McKay b194b7df0d
Add the ability to use a custom allocator for fetches to avoid unnecessary copies in control flow operators. (#377)
* Add the ability to use a custom allocator for fetches.

Allows control flow nodes to forward the allocation to the control flow op and avoid an unnecessary copy when the subgraph output has a symbolic dimension.

Update Scan and If to use custom allocators when applicable.

* Remove unnecessary forward declaration

* Fix Mac build warnings
2019-01-29 19:48:10 +10:00
..
cuda Abbreviate ONNXRuntime as Ort in all of our public APIs (#175) 2018-12-14 14:54:23 -08:00
inference_session_without_environment Build system upgrades (#281) 2019-01-07 13:15:24 -08:00
allocation_planner_test.cc cleanup code (#343) 2019-01-16 17:12:22 -08:00
allocator_test.cc Abbreviate ONNXRuntime as Ort in all of our public APIs (#175) 2018-12-14 14:54:23 -08:00
bfc_arena_test.cc Initial bootstrap commit. 2018-11-19 16:48:22 -08:00
data_types_test.cc Add a missing header (#243) 2018-12-21 19:06:09 -08:00
execution_frame_test.cc Add the ability to use a custom allocator for fetches to avoid unnecessary copies in control flow operators. (#377) 2019-01-29 19:48:10 +10:00
float_16_test.cc Abbreviate ONNXRuntime as Ort in all of our public APIs (#175) 2018-12-14 14:54:23 -08:00
header_files_test.cc Initial bootstrap commit. 2018-11-19 16:48:22 -08:00
inference_session_test.cc Refactor InferenceSession::Impl::Load code to remove duplication. (#248) 2019-01-29 17:18:38 +10:00
insert_cast_transformer_test.cc Document the Graph header files and cleanup some issues. (#42) 2018-11-28 08:42:11 -08:00
local_kernel_registry_test.cc Abbreviate ONNXRuntime as Ort in all of our public APIs (#175) 2018-12-14 14:54:23 -08:00
math_test.cc Initial bootstrap commit. 2018-11-19 16:48:22 -08:00
mem_pattern_planner_test.cc Initial bootstrap commit. 2018-11-19 16:48:22 -08:00
memcpy_transformer_test.cc Document the Graph header files and cleanup some issues. (#42) 2018-11-28 08:42:11 -08:00
model_builder_utils.h Initial bootstrap commit. 2018-11-19 16:48:22 -08:00
op_kernel_test.cc Abbreviate ONNXRuntime as Ort in all of our public APIs (#175) 2018-12-14 14:54:23 -08:00
opaque_kernels_test.cc Abbreviate ONNXRuntime as Ort in all of our public APIs (#175) 2018-12-14 14:54:23 -08:00
session_state_test.cc Abbreviate ONNXRuntime as Ort in all of our public APIs (#175) 2018-12-14 14:54:23 -08:00
shape_inference_test.cc Document the Graph header files and cleanup some issues. (#42) 2018-11-28 08:42:11 -08:00
tensor_test.cc Abbreviate ONNXRuntime as Ort in all of our public APIs (#175) 2018-12-14 14:54:23 -08:00
tensorprotoutils_test.cc Abbreviate ONNXRuntime as Ort in all of our public APIs (#175) 2018-12-14 14:54:23 -08:00
tensorutils_test.cc Initial bootstrap commit. 2018-11-19 16:48:22 -08:00
test_main.cc Initial bootstrap commit. 2018-11-19 16:48:22 -08:00
test_utils.cc trt (#361) 2019-01-23 13:28:13 -08:00
test_utils.h trt (#361) 2019-01-23 13:28:13 -08:00
TestAllocatorManager.cc Abbreviate ONNXRuntime as Ort in all of our public APIs (#175) 2018-12-14 14:54:23 -08:00
TestAllocatorManager.h Abbreviate ONNXRuntime as Ort in all of our public APIs (#175) 2018-12-14 14:54:23 -08:00