Home
last modified time | relevance | path

Searched refs:num_retvals (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc153 int num_retvals = 1; in BM_Execute() local
156 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in BM_Execute()
359 int num_retvals = 1; in TEST() local
360 TFE_Execute(matmul, &retvals[0], &num_retvals, status.get()); in TEST()
400 int num_retvals = 1; in TEST() local
401 TFE_Execute(matmul, &retvals[0], &num_retvals, status.get()); in TEST()
455 int num_retvals = 2; // Should be reduced to 1 by the TFE_Execute call. in TEST() local
456 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in TEST()
462 ASSERT_EQ(1, num_retvals); in TEST()
489 int num_retvals = 2; // Should be reduced to 1 by the TFE_Execute call. in TEST() local
[all …]
Dc_api.cc777 void TFE_Execute(TFE_Op* op, TFE_TensorHandle** retvals, int* num_retvals, in TFE_Execute() argument
879 *num_retvals = std::min<int>(*num_retvals, outputs.size()); in TFE_Execute()
880 for (int i = 0; i < *num_retvals; ++i) { in TFE_Execute()
Dc_api.h249 int* num_retvals, TF_Status* status);
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc1710 int num_retvals = 0; local
1714 num_retvals += attr_list_sizes[output_arg.number_attr()];
1716 num_retvals += attr_list_sizes[output_arg.type_list_attr()];
1718 num_retvals++;
1722 tensorflow::gtl::InlinedVector<TFE_TensorHandle*, 2> retvals(num_retvals);
1725 TFE_Execute(op, retvals.data(), &num_retvals, status);
1739 PyObject* flat_result = PyList_New(num_retvals);
1740 for (int i = 0; i < num_retvals; ++i) {
/external/tensorflow/tensorflow/core/framework/
Dfunction.h243 virtual size_t num_retvals() const = 0;
266 size_t num_retvals() const override { return ret_types_.size(); } in num_retvals() function
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.cc61 size_t num_retvals() const override { return retvals_.size(); } in num_retvals() function in tensorflow::__anonf8786f2a0111::CallFrameBase