onnxruntime/onnxruntime/core/framework
Sherlock 2a02871157
Disable reuse for YieldOp's inputs (FW partial graph's output) (#7767)
* Disable reuse for YieldOp's input

Co-authored-by: Sherlock Huang <bahuang@OrtTrainingDev3.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>
2021-05-20 21:39:36 -07:00
..
allocation_planner.cc Disable reuse for YieldOp's inputs (FW partial graph's output) (#7767) 2021-05-20 21:39:36 -07:00
allocation_planner.h Flag for tensor memory re-use in allocation planner. (#7359) 2021-04-16 17:53:25 -07:00
allocator.cc Share allocator between CUDA EP & TRT EP. (#6332) 2021-01-27 00:14:43 -08:00
allocatormgr.cc Some cosmetic changes (#7741) 2021-05-18 00:02:07 -07:00
allocatormgr.h Add ability for memory arenas to "shrink" periodically (#7284) 2021-05-08 07:53:21 -07:00
arena.h Add ability for memory arenas to "shrink" periodically (#7284) 2021-05-08 07:53:21 -07:00
arena_extend_strategy.h Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
bfc_arena.cc Some cosmetic changes (#7741) 2021-05-18 00:02:07 -07:00
bfc_arena.h Some cosmetic changes (#7741) 2021-05-18 00:02:07 -07:00
callback.cc
callback.h
compute_capability.h
config_options.cc Add ability for memory arenas to "shrink" periodically (#7284) 2021-05-08 07:53:21 -07:00
config_options.h Add ability for memory arenas to "shrink" periodically (#7284) 2021-05-08 07:53:21 -07:00
customregistry.cc Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
data_transfer.cc
data_transfer.h Make IDataTransfer be directly shared with shared providers (#7215) 2021-04-01 20:39:16 -07:00
data_transfer_manager.cc
data_transfer_manager.h
data_transfer_utils.h Remove conditional compilation of std::is_trivially_copyable since we are no longer supporting GCC 4. (#7504) 2021-04-29 19:13:09 -07:00
data_types.cc Enable type reduction for Shrink, Sign, SplitToSequence CPU kernels (#7090) 2021-03-23 09:57:33 -07:00
debug_node_inputs_outputs_utils.cc Fix DEBUG_NODE_INPUTS_OUTPUTS test by putting it in a separate process, clean up unused test_main.cc files. (#5949) 2020-12-11 11:36:58 -08:00
debug_node_inputs_outputs_utils.h Allow turning off printing of shape when compiled with onnxruntime_DEBUG_NODE_INPUTS_OUTPUTS. (#5768) 2020-11-11 18:59:04 -08:00
element_type_lists.h Enable type reduction for Shrink, Sign, SplitToSequence CPU kernels (#7090) 2021-03-23 09:57:33 -07:00
empty.cc
endian_utils.cc Reduce tensorprotoutils binary size (#6634) 2021-02-12 16:48:13 +10:00
endian_utils.h Remove conditional compilation of std::is_trivially_copyable since we are no longer supporting GCC 4. (#7504) 2021-04-29 19:13:09 -07:00
error_code.cc
error_code_helper.h
ex_lib_loader.cc
ex_lib_loader.h
execution_frame.cc Add ATenOp and call aten::embedding and its Backward Op from ORT (#7590) 2021-05-13 09:24:27 +08:00
execution_frame.h Add ATenOp and call aten::embedding and its Backward Op from ORT (#7590) 2021-05-13 09:24:27 +08:00
execution_plan_base.h
execution_provider.cc Change onnxruntime::make_unique to std::make_unique (#7502) 2021-04-29 17:04:53 -07:00
execution_providers.h Release buffers for prepacked tensors (#6820) 2021-03-10 14:07:20 -08:00
fallback_cpu_capability.cc Revert (#7663) 2021-05-13 14:11:17 -07:00
fallback_cpu_capability.h Reduce amount of compiled CUDA device code (#6118) 2020-12-14 15:27:40 -08:00
feeds_fetches_manager.cc Change onnxruntime::make_unique to std::make_unique (#7502) 2021-04-29 17:04:53 -07:00
feeds_fetches_manager.h Ryanunderhill/cuda shared (#7626) 2021-05-20 07:53:47 -07:00
func_kernel.cc
func_kernel.h Support EPs that compile nodes in a minimal build. (#5776) 2020-11-17 13:52:22 +10:00
fuse_nodes_funcs.cc Support EPs that compile nodes in a minimal build. (#5776) 2020-11-17 13:52:22 +10:00
fuse_nodes_funcs.h Change onnxruntime::make_unique to std::make_unique (#7502) 2021-04-29 17:04:53 -07:00
graph_partitioner.cc Change onnxruntime::make_unique to std::make_unique (#7502) 2021-04-29 17:04:53 -07:00
graph_partitioner.h Support EPs that compile nodes in a minimal build. (#5776) 2020-11-17 13:52:22 +10:00
iexecutor.h
kernel_def_builder.cc Check kernel def hashes (#7120) 2021-04-01 17:42:58 -07:00
kernel_registry.cc Change onnxruntime::make_unique to std::make_unique (#7502) 2021-04-29 17:04:53 -07:00
kernel_registry_manager.cc Require explicit inclusion of custom op support in a minimal build (#6663) 2021-02-13 12:42:33 +10:00
kernel_registry_manager.h Require explicit inclusion of custom op support in a minimal build (#6663) 2021-02-13 12:42:33 +10:00
mem_buffer.h
mem_pattern.h Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
mem_pattern_planner.h Exclude some training specific code from the minimal build. Cleanup some related aspects of allocation planner. (#5861) 2020-11-20 20:25:46 +10:00
memcpy.cc Rename MakeString and ParseString functions. (#6272) 2021-01-07 15:43:42 -08:00
memcpy.h
memory_info.cc Add New AllocKind for YieldOp Outputs, Run YieldOp with InferenceSession in UT (#7125) 2021-03-25 15:18:51 +08:00
memory_info.h Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
mimalloc_arena.cc
mimalloc_arena.h
mldata_type_utils.cc
mldata_type_utils.h
murmurhash3.cc
murmurhash3.h
node_index_info.cc
node_index_info.h Factor out IAllocator so that it can be shared with shared providers (#5567) 2020-10-27 17:28:17 -07:00
onnxruntime_map_type_info.cc
onnxruntime_map_type_info.h
onnxruntime_sequence_type_info.cc
onnxruntime_sequence_type_info.h
onnxruntime_typeinfo.cc Add LearningModelBuilder to WinML Experimental Namespace along with various Audio operators (#6623) 2021-02-12 14:17:10 -08:00
onnxruntime_typeinfo.h Add LearningModelBuilder to WinML Experimental Namespace along with various Audio operators (#6623) 2021-02-12 14:17:10 -08:00
op_kernel.cc Add ATenOp and call aten::embedding and its Backward Op from ORT (#7590) 2021-05-13 09:24:27 +08:00
op_kernel_context_internal.h Address comments from ORTModule master merge (#7101) 2021-03-26 16:26:42 -07:00
op_kernel_info.cc Support EPs that compile nodes in a minimal build. (#5776) 2020-11-17 13:52:22 +10:00
op_node_proto_helper.cc Allow Kernels refer to some attribute data directly in the protobuf (#5624) 2020-10-29 16:12:54 -07:00
ort_stl_allocator.h
ort_value_name_idx_map.h
ort_value_pattern_planner.cc Change onnxruntime::make_unique to std::make_unique (#7502) 2021-04-29 17:04:53 -07:00
ort_value_pattern_planner.h Exclude some training specific code from the minimal build. Cleanup some related aspects of allocation planner. (#5861) 2020-11-20 20:25:46 +10:00
ort_value_tensor_slicer.cc Ryanunderhill/cuda shared (#7626) 2021-05-20 07:53:47 -07:00
ort_value_tensor_slicer.h Ryanunderhill/cuda shared (#7626) 2021-05-20 07:53:47 -07:00
orttraining_partial_executor.cc Revert CUPTI profiling feature (#7763) 2021-05-19 21:54:29 -07:00
orttraining_partial_executor.h Partial graph execution made simple. (#7324) 2021-04-23 15:09:18 -07:00
parallel_executor.cc Revert CUPTI profiling feature (#7763) 2021-05-19 21:54:29 -07:00
parallel_executor.h
partial_graph_execution_state.h Change onnxruntime::make_unique to std::make_unique (#7502) 2021-04-29 17:04:53 -07:00
prepacked_weights.cc Add ability for pre-packed weights of shared initializers to be shared across sessions (#7421) 2021-05-14 20:44:42 -07:00
prepacked_weights.h Add ability for pre-packed weights of shared initializers to be shared across sessions (#7421) 2021-05-14 20:44:42 -07:00
prepacked_weights_container.cc Fix builds that use gcc 5 (#7765) 2021-05-20 09:19:04 -07:00
prepacked_weights_container.h Add ability for pre-packed weights of shared initializers to be shared across sessions (#7421) 2021-05-14 20:44:42 -07:00
provider_bridge_ort.cc Ryanunderhill/cuda shared (#7626) 2021-05-20 07:53:47 -07:00
provider_bridge_ort.h Ryanunderhill/cuda shared (#7626) 2021-05-20 07:53:47 -07:00
random_generator.cc
random_generator.h
random_seed.cc
random_seed.h
run_options.cc Add ability for memory arenas to "shrink" periodically (#7284) 2021-05-08 07:53:21 -07:00
sequential_execution_plan.h Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
sequential_executor.cc Revert CUPTI profiling feature (#7763) 2021-05-19 21:54:29 -07:00
sequential_executor.h
session_options.cc Add ability for memory arenas to "shrink" periodically (#7284) 2021-05-08 07:53:21 -07:00
session_options.h Add ability for memory arenas to "shrink" periodically (#7284) 2021-05-08 07:53:21 -07:00
session_state.cc Add ability for pre-packed weights of shared initializers to be shared across sessions (#7421) 2021-05-14 20:44:42 -07:00
session_state.h Add ability for pre-packed weights of shared initializers to be shared across sessions (#7421) 2021-05-14 20:44:42 -07:00
session_state_utils.cc Add ability for memory arenas to "shrink" periodically (#7284) 2021-05-08 07:53:21 -07:00
session_state_utils.h Release buffers for prepacked tensors (#6820) 2021-03-10 14:07:20 -08:00
simple_tensor_allocator.cc Release buffers for prepacked tensors (#6820) 2021-03-10 14:07:20 -08:00
simple_tensor_allocator.h Release buffers for prepacked tensors (#6820) 2021-03-10 14:07:20 -08:00
sparse_tensor.cc Support for Sparse Initializers (#5540) 2020-10-27 10:32:06 -07:00
tensor.cc Add ability to allocate initialized tensor memory from non-arena memory (#7267) 2021-04-20 20:27:48 -07:00
tensor_allocator.cc Change onnxruntime::make_unique to std::make_unique (#7502) 2021-04-29 17:04:53 -07:00
tensor_allocator.h Release buffers for prepacked tensors (#6820) 2021-03-10 14:07:20 -08:00
tensor_allocator_with_mem_pattern.h Change onnxruntime::make_unique to std::make_unique (#7502) 2021-04-29 17:04:53 -07:00
tensor_external_data_info.cc Change onnxruntime::make_unique to std::make_unique (#7502) 2021-04-29 17:04:53 -07:00
tensor_external_data_info.h
tensor_shape.cc
tensor_type_and_shape.cc Initial Objective-C API (#7366) 2021-04-27 10:06:30 -07:00
tensor_type_and_shape.h Initial Objective-C API (#7366) 2021-04-27 10:06:30 -07:00
tensorprotoutils.cc Remove conditional compilation of std::is_trivially_copyable since we are no longer supporting GCC 4. (#7504) 2021-04-29 19:13:09 -07:00
tensorprotoutils.h Ryanunderhill/cuda shared (#7626) 2021-05-20 07:53:47 -07:00
TensorSeq.h Support opset-13 specs of controlflow ops (Loop, If) (#5665) 2020-11-11 23:44:14 -08:00
utils.cc Change onnxruntime::make_unique to std::make_unique (#7502) 2021-04-29 17:04:53 -07:00
utils.h Partial graph execution made simple. (#7324) 2021-04-23 15:09:18 -07:00