Searched refs:ObtainKernelOutputs (Results 1 – 4 of 4) sorted by relevance
59 std::vector<kernel::AddressPtr> LaunchKernel::ObtainKernelOutputs(const std::vector<size_t> &output… in ObtainKernelOutputs() function in mindspore::device::LaunchKernel82 auto kernel_outputs = ObtainKernelOutputs(kernel_mod_->GetOutputSizeList()); in LaunchSingleKernel()
53 std::vector<kernel::AddressPtr> ObtainKernelOutputs(const std::vector<size_t> &outputs_list);
62 auto kernel_outputs = ObtainKernelOutputs(kernel_mod_->GetOutputSizeList()); in LaunchOpKernel()
64 auto kernel_outputs = ObtainKernelOutputs(kernel_mod_->GetOutputSizeList()); in LaunchOpKernel()