Function get_tensor_mem_size

executor/lib/tensor_mem.cpp:43–54

}

int get_tensor_mem_size(const Tensor* tensor)
{
    if(tensor->GetType() == kConstTensor)
        return tensor->GetTotalSize();

    TensorMemPtr ptr;

    if(get_tensor_memptr(tensor, ptr) && ptr.get() != nullptr)
        return ptr.get()->GetSize();
    else
        return 0;
}

bool get_tensor_memptr(const Tensor* tensor, TensorMemPtr& ptr)
{