Searched refs:num_return_outputs (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/c/
H A Dc_api.cc2070 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 Dc_api.h1002 // `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 Dc_api_test.cc724 int num_return_outputs; local
725 TF_ImportGraphDefResultsReturnOutputs(results, &num_return_outputs,
727 ASSERT_EQ(2, num_return_outputs);

Completed in 82 milliseconds