Searched refs:SetOutputExpression (Results 1 – 3 of 3) sorted by relevance
603 void XlaOpKernelContext::SetOutputExpression(int index, in SetOutputExpression() function in tensorflow::XlaOpKernelContext649 SetOutputExpression( in SetOutput()655 SetOutputExpression(index, XlaExpression::Constant(constant)); in SetConstantOutput()660 SetOutputExpression(index, XlaExpression::TensorList(handle)); in SetTensorListOutput()664 SetOutputExpression(index, XlaExpression::Resource(resource)); in SetResourceOutput()
238 void SetOutputExpression(int index, const XlaExpression& expression);
104 ctx->SetOutputExpression(0, arg); in Compile()