Searched defs:GetPrevNodeOutput (Results 1 – 2 of 2) sorted by relevance
1427 session::KernelWithIndex GPUKernelRuntime::GetPrevNodeOutput(const AnfNodePtr &node, size_t i) { in GetPrevNodeOutput() function in mindspore::device::gpu::GPUKernelRuntime
656 KernelWithIndex AnfAlgo::GetPrevNodeOutput(const AnfNodePtr &anf_node, size_t input_idx, bool skip_… in GetPrevNodeOutput() function in mindspore::common::AnfAlgo