Searched refs:regression_input (Results 1 – 1 of 1) sorted by relevance
171 const Tensor& regression_input, Session* session, in RunRegression() argument180 session->Run({{signature.input().tensor_name(), regression_input}}, in RunRegression()195 TF_RETURN_IF_ERROR(BatchSizesMatch(regression_input, *regression_output)); in RunRegression()