Searched refs:regression_output (Results 1 – 2 of 2) sorted by relevance
100 regression_output = estimator_spec.export_outputs["regression_head"]102 regression_output, core_export_lib.RegressionOutput))104 regression_output.value.eval())120 regression_output = estimator_spec.export_outputs["regression_head"]122 regression_output, core_export_lib.RegressionOutput))124 regression_output.value.eval())283 regression_output = estimator_spec.export_outputs["regression_head"]285 regression_output, core_export_lib.RegressionOutput))287 regression_output.value.eval())
172 Tensor* regression_output) { in RunRegression() argument174 if (regression_output) { in RunRegression()193 if (regression_output) { in RunRegression()194 *regression_output = outputs[0]; in RunRegression()195 TF_RETURN_IF_ERROR(BatchSizesMatch(regression_input, *regression_output)); in RunRegression()