MCPcopy Create free account
hub / github.com/OAID/Tengine / get_tensor_memptr

Function get_tensor_memptr

executor/lib/tensor_mem.cpp:56–65  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

54}
55
56bool get_tensor_memptr(const Tensor* tensor, TensorMemPtr& ptr)
57{
58 if(tensor->ExistAttr("tensor_mem"))
59 {
60 ptr = any_cast<TensorMemPtr>(tensor->GetAttr("tensor_mem"));
61 return true;
62 }
63
64 return false;
65}
66
67bool set_tensor_mem(Tensor* tensor, void* addr, int size, mem_release_t releaser)
68{

Callers 2

get_tensor_memFunction · 0.85
get_tensor_mem_sizeFunction · 0.85

Calls 2

ExistAttrMethod · 0.45
GetAttrMethod · 0.45

Tested by

no test coverage detected