Searched refs:num_retvals (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/c/eager/ |
D | c_api_test.cc | 153 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 …]
|
D | c_api.cc | 777 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()
|
D | c_api.h | 249 int* num_retvals, TF_Status* status);
|
/external/tensorflow/tensorflow/python/eager/ |
D | pywrap_tfe_src.cc | 1710 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/ |
D | function.h | 243 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/ |
D | captured_function.cc | 61 size_t num_retvals() const override { return retvals_.size(); } in num_retvals() function in tensorflow::__anonf8786f2a0111::CallFrameBase
|