Searched refs:IsGraphOutput (Results 1 – 8 of 8) sorted by relevance
84 if (tensor->IsGraphOutput()) { in WaitLockedInputs()94 if (input_tensor->NeedWait() && !input_tensor->IsGraphOutput()) { in WaitLockedInputs()
1682 if (!tensor->IsGraphOutput()) { in GetInputNeedLockTensors()
76 bool IsGraphOutput(const session::KernelGraph *graph, const mindspore::AnfNodePtr &kernel) const;
657 bool GPUKernelRuntime::IsGraphOutput(const session::KernelGraph *graph, const mindspore::AnfNodePtr… in IsGraphOutput() function in mindspore::device::gpu::GPUKernelRuntime678 if (IsGraphOutput(graph, kernel)) { in ClearKernelOutputAddress()
369 bool IsGraphOutput() { return graph_output_; } in IsGraphOutput() function
126 if (out_kernel_out_tensor->IsGraphOutput()) { in SubgraphOutputTensors()
186 bool IsGraphOutput() const { return this->category_ == GRAPH_OUTPUT; } in IsGraphOutput() function
825 if (tensor->IsGraphOutput() && tensor->allocator() == nullptr) { in ~LiteSession()