Searched refs:num_return_outputs (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/c/ |
H A D | c_api.cc | 2070 int num_return_outputs, TF_Status* status) { 2071 if (num_return_outputs != options->opts.return_tensors.size()) { 2072 status->status = InvalidArgument("Expected 'num_return_outputs' to be ", 2074 ", got ", num_return_outputs); 2077 if (num_return_outputs > 0 && return_outputs == nullptr) { 2079 "'return_outputs' must be preallocated to length ", num_return_outputs); 2090 DCHECK_EQ(results.return_tensors.size(), num_return_outputs); 2092 num_return_outputs * sizeof(TF_Output)); 2067 TF_GraphImportGraphDefWithReturnOutputs( TF_Graph* graph, const TF_Buffer* graph_def, const TF_ImportGraphDefOptions* options, TF_Output* return_outputs, int num_return_outputs, TF_Status* status) argument
|
H A D | c_api.h | 1002 // `num_return_outputs` must be the number of return outputs added (i.e. the 1004 // `num_return_outputs` is non-zero, `return_outputs` must be of length 1005 // `num_return_outputs`. Otherwise it can be null. 1009 int num_return_outputs, TF_Status* status);
|
H A D | c_api_test.cc | 724 int num_return_outputs; local 725 TF_ImportGraphDefResultsReturnOutputs(results, &num_return_outputs, 727 ASSERT_EQ(2, num_return_outputs);
|
Completed in 82 milliseconds