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