Searched refs:workspaces_addr_ (Results 1 – 5 of 5) sorted by relevance
72 workspaces_addr_ = std::vector<uint8_t *>(workspaces_list.size(), nullptr); in ObtainKernelWorkspaces()73 kernel_workspace = ObtainKernelAddress(workspaces_list, &workspaces_addr_); in ObtainKernelWorkspaces()100 for (size_t i = 0; i < workspaces_addr_.size(); ++i) { in FreeOutputAndWorkspaceDeviceMem()101 if (workspaces_addr_[i] != nullptr) { in FreeOutputAndWorkspaceDeviceMem()102 FreeDeviceMem(workspaces_addr_[i]); in FreeOutputAndWorkspaceDeviceMem()103 workspaces_addr_[i] = nullptr; in FreeOutputAndWorkspaceDeviceMem()107 workspaces_addr_.clear(); in FreeOutputAndWorkspaceDeviceMem()
48 std::vector<uint8_t *> workspaces_addr_; variable
101 auto item = workspaces_addr_.find(variable);102 if (item != workspaces_addr_.end()) {149 std::map<void *, std::string> workspaces_addr_; variable
77 …workspaces_addr_.insert(std::make_pair(addr, kBufferPrefixNameAdd + std::to_string(tensors_size_ +… in AssignWorkspaces()
193 void set_workspaces_addr(const std::vector<AddressPtr> &addr) { workspaces_addr_ = addr; } in set_workspaces_addr()196 const std::vector<AddressPtr> &GetWorkSpacesAddr() { return workspaces_addr_; } in GetWorkSpacesAddr()210 std::vector<AddressPtr> workspaces_addr_; variable