Searched refs:num_return_outputs (Results 1 – 3 of 3) sorted by relevance
1781 int num_return_outputs, TF_Status* status) { in TF_GraphImportGraphDefWithReturnOutputs() argument1782 if (num_return_outputs != options->opts.return_tensors.size()) { in TF_GraphImportGraphDefWithReturnOutputs()1785 ", got ", num_return_outputs); in TF_GraphImportGraphDefWithReturnOutputs()1788 if (num_return_outputs > 0 && return_outputs == nullptr) { in TF_GraphImportGraphDefWithReturnOutputs()1790 "'return_outputs' must be preallocated to length ", num_return_outputs); in TF_GraphImportGraphDefWithReturnOutputs()1802 DCHECK_EQ(results.return_tensors.size(), num_return_outputs); in TF_GraphImportGraphDefWithReturnOutputs()1804 num_return_outputs * sizeof(TF_Output)); in TF_GraphImportGraphDefWithReturnOutputs()
854 int num_return_outputs, TF_Status* status);
782 int num_return_outputs; in TEST() local783 TF_ImportGraphDefResultsReturnOutputs(results, &num_return_outputs, in TEST()785 ASSERT_EQ(2, num_return_outputs); in TEST()