Searched refs:num_return_outputs (Results 1 – 3 of 3) sorted by relevance
2188 int num_return_outputs, TF_Status* status) { in TF_GraphImportGraphDefWithReturnOutputs() argument2189 if (num_return_outputs != options->opts.return_tensors.size()) { in TF_GraphImportGraphDefWithReturnOutputs()2192 ", got ", num_return_outputs); in TF_GraphImportGraphDefWithReturnOutputs()2195 if (num_return_outputs > 0 && return_outputs == nullptr) { in TF_GraphImportGraphDefWithReturnOutputs()2197 "'return_outputs' must be preallocated to length ", num_return_outputs); in TF_GraphImportGraphDefWithReturnOutputs()2209 DCHECK_EQ(results.return_tensors.size(), num_return_outputs); in TF_GraphImportGraphDefWithReturnOutputs()2211 num_return_outputs * sizeof(TF_Output)); in TF_GraphImportGraphDefWithReturnOutputs()
1058 int num_return_outputs, TF_Status* status);
745 int num_return_outputs; in TEST() local746 TF_ImportGraphDefResultsReturnOutputs(results, &num_return_outputs, in TEST()748 ASSERT_EQ(2, num_return_outputs); in TEST()