13 namespace onnxruntime {
15 class OrtValueNameIdxMap;
17 class DataTransferManager;
18 struct AllocPlanPerValue;
28 const std::unordered_map<int, OrtValue>& constant_initialized_tensors,
29 const OrtValueNameIdxMap& mlvalue_name_idx_map,
30 const DataTransferManager& data_transfer_mgr);
56 gsl::not_null<const ::onnxruntime::IExecutionProvider*> execution_provider_;
57 const std::unordered_map<int, OrtValue>& constant_initialized_tensors_;
58 const OrtValueNameIdxMap& ort_value_name_idx_map_;
59 const DataTransferManager& data_transfer_mgr_;
const IExecutionProvider * GetExecutionProvider() const noexcept
const DataTransferManager & GetDataTransferManager() const noexcept
OpKernelInfo(const onnxruntime::Node &node, const KernelDef &kernel_def, const IExecutionProvider &execution_provider, const std::unordered_map< int, OrtValue > &constant_initialized_tensors, const OrtValueNameIdxMap &mlvalue_name_idx_map, const DataTransferManager &data_transfer_mgr)
const onnxruntime::Node & node() const noexcept
AllocatorPtr GetAllocator(int device_id, OrtMemType mem_type) const
const OrtMemoryInfo & GetMemoryInfo(int device_id, OrtMemType mem_type) const
std::shared_ptr< IAllocator > AllocatorPtr
bool TryGetConstantInput(int input_index, const Tensor **constant_input_value) const
const KernelDef & GetKernelDef() const
OrtMemType
Memory types for allocated memory, execution provider specific types should be extended in each provi...