Searched defs:SetWorkspaceAddr (Results 1 – 3 of 3) sorted by relevance
91 bool KernelInfo::SetWorkspaceAddr(const DeviceAddressPtr &output_address, size_t index) { in SetWorkspaceAddr() function in mindspore::device::KernelInfo
570 TEST_F(AnfRuntimeAlgorithmTest, SetWorkspaceAddr) { in TEST_F() argument
1127 void AnfRuntimeAlgorithm::SetWorkspaceAddr(const DeviceAddressPtr &addr, size_t output_idx, AnfNode… in SetWorkspaceAddr() function in mindspore::session::AnfRuntimeAlgorithm