Searched refs:SetResourceOutput (Results 1 – 5 of 5) sorted by relevance
48 context->SetResourceOutput(0, resource.get()); in Compile()
208 void SetResourceOutput(int index, XlaResource* resource);
603 void XlaOpKernelContext::SetResourceOutput(int index, XlaResource* resource) { in SetResourceOutput() function in tensorflow::XlaOpKernelContext
113 ctx->SetResourceOutput(0, resource); in Compile()
173 ctx->SetResourceOutput(0, var); in Compile()594 ctx->SetResourceOutput(0, gradient); in Compile()