Searched refs:BatchSizesMatch (Results 1 – 1 of 1) sorted by relevance
38 Status BatchSizesMatch(const Tensor& input, const Tensor& output) { in BatchSizesMatch() function143 TF_RETURN_IF_ERROR(BatchSizesMatch(input, *classes)); in RunClassification()147 TF_RETURN_IF_ERROR(BatchSizesMatch(input, *scores)); in RunClassification()195 TF_RETURN_IF_ERROR(BatchSizesMatch(regression_input, *regression_output)); in RunRegression()