Home
last modified time | relevance | path

Searched refs:base_mem_storage_ (Results 1 – 2 of 2) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/
H A Dmemory_tracking.hpp458 , base_mem_storage_(base_mem_storage) in grantor_t()
463 , base_mem_storage_(parent.base_mem_storage_) in grantor_t()
468 if (!base_mem_storage_) { in get()
475 char *host_storage_ptr = get_host_storage_ptr(base_mem_storage_); in get()
476 char *base_ptr = host_storage_ptr + base_mem_storage_->base_offset(); in get()
482 if (!base_mem_storage_) { in get_memory_storage()
489 if (is_cpu_engine(base_mem_storage_)) { in get_memory_storage()
492 = host_storage_ptr + base_mem_storage_->base_offset(); in get_memory_storage()
502 return base_mem_storage_->get_sub_storage(aligned_offset, e.size); in get_memory_storage()
506 return base_mem_storage_; in get_base_storage()
[all …]
/dports/math/onednn/oneDNN-2.5.1/src/common/
H A Dmemory_tracking.hpp459 , base_mem_storage_(base_mem_storage) in grantor_t()
464 , base_mem_storage_(parent.base_mem_storage_) in grantor_t()
469 if (!base_mem_storage_) { in get()
476 char *host_storage_ptr = get_host_storage_ptr(base_mem_storage_); in get()
477 char *base_ptr = host_storage_ptr + base_mem_storage_->base_offset(); in get()
483 if (!base_mem_storage_) { in get_memory_storage()
490 if (is_cpu_engine(base_mem_storage_)) { in get_memory_storage()
493 = host_storage_ptr + base_mem_storage_->base_offset(); in get_memory_storage()
503 return base_mem_storage_->get_sub_storage(aligned_offset, e.size); in get_memory_storage()
507 return base_mem_storage_; in get_base_storage()
[all …]