Searched refs:GetTensorBuffer (Results 1 – 2 of 2) sorted by relevance
1684 inline void* GetTensorBuffer(const Tensor* tensor) const { in GetTensorBuffer() function1711 SetUsrMem(dim, fm, GetTensorBuffer(tensor)); in SetUsrMem()1743 SetUsrMem(dim, strides, GetTensorBuffer(tensor)); in SetUsrMem()1758 SetUsrMem(md, GetTensorBuffer(tensor)); in SetUsrMem()1781 SetUsrMem(pd, GetTensorBuffer(tensor)); in SetUsrMem()1817 user_memory_->set_data_handle(GetTensorBuffer(tensor)); in SetUsrMemDataHandle()1988 return CheckReorderToOpMem(op_pd, GetTensorBuffer(reorder_tensor), net); in CheckReorderToOpMem()1997 return CheckReorderToOpMem(op_pd, GetTensorBuffer(reorder_tensor)); in CheckReorderToOpMem()
1135 filter.GetTensorBuffer(filter_out_tensor)); in Compute()1372 filter->GetTensorBuffer(filter_out_tensor)); in PrepareAndExecuteNet()1424 void* cached_filter_data = filter.GetTensorBuffer(filter_tensor_ptr); in CacheFilter()