Searched refs:SetOutputExpression (Results 1 – 3 of 3) sorted by relevance
436 void XlaOpKernelContext::SetOutputExpression(int index, in SetOutputExpression() function in tensorflow::XlaOpKernelContext482 SetOutputExpression( in SetOutput()488 SetOutputExpression(index, XlaExpression::Constant(constant)); in SetConstantOutput()493 SetOutputExpression(index, XlaExpression::TensorList(handle)); in SetTensorListOutput()497 SetOutputExpression(index, XlaExpression::Resource(resource)); in SetResourceOutput()
178 void SetOutputExpression(int index, const XlaExpression& expression);
59 ctx->SetOutputExpression(0, arg); in Compile()