Searched refs:SetResourceOutput (Results 1 – 5 of 5) sorted by relevance
45 context->SetResourceOutput(0, resource.get()); in Compile()
193 void SetResourceOutput(int index, XlaResource* resource);
496 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()