| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test.cc | 73 int num_retvals = 1; in BM_Execute() local
76 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in BM_Execute()
175 int num_retvals = 1; in TestRemoteExecute() local
176 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in TestRemoteExecute()
257 int num_retvals = 1; in TestRemoteExecuteSilentCopies() local
258 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in TestRemoteExecuteSilentCopies()
328 int num_retvals = 1; in CheckRemoteMatMulExecutesOK() local
329 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in CheckRemoteMatMulExecutesOK()
677 int num_retvals = 1; in TensorHandleSilentCopy() local
678 TFE_Execute(matmul, &retvals[0], &num_retvals, status.get()); in TensorHandleSilentCopy()
[all …]
|
| D | c_api_experimental_test.cc | 54 int num_retvals = 1; in ExecuteWithProfiling() local
65 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in ExecuteWithProfiling()
71 ASSERT_EQ(1, num_retvals); in ExecuteWithProfiling()
|
| D | c_api.cc | 810 void TFE_Execute(TFE_Op* op, TFE_TensorHandle** retvals, int* num_retvals, in TFE_Execute() argument
814 *num_retvals); in TFE_Execute()
816 tensorflow::EagerExecute(&op->operation, &handle_retvals, num_retvals); in TFE_Execute()
820 for (int i = 0; i < *num_retvals; ++i) { in TFE_Execute()
|
| D | c_api.h | 387 int* num_retvals, TF_Status* status);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 48 int* num_retvals) { in GetNumRetvals() argument
66 *num_retvals += iter->second.i(); in GetNumRetvals()
74 *num_retvals += iter->second.list().type_size(); in GetNumRetvals()
76 *num_retvals += 1; in GetNumRetvals()
185 int num_retvals = 0; in ExecuteOp() local
188 operation.attrs(), &num_retvals)); in ExecuteOp()
191 TF_RETURN_IF_ERROR(EagerExecute(op.get(), &retvals, &num_retvals)); in ExecuteOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 368 int* num_retvals) { in EagerLocalExecute() argument
466 if (output_dtypes_size > *num_retvals) { in EagerLocalExecute()
469 *num_retvals); in EagerLocalExecute()
471 *num_retvals = output_dtypes_size; in EagerLocalExecute()
498 retvals->resize(*num_retvals); in EagerLocalExecute()
505 for (int i = 0; i < *num_retvals; ++i) { in EagerLocalExecute()
521 retvals->data(), *num_retvals); in EagerLocalExecute()
641 int* num_retvals) { in EagerRemoteExecute() argument
700 if (*num_retvals != output_dtypes.size()) { in EagerRemoteExecute()
715 for (int i = 0; i < *num_retvals; i++) { in EagerRemoteExecute()
[all …]
|
| D | execute.h | 42 int* num_retvals);
51 TensorHandle** retvals, int num_retvals);
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 191 int num_retvals = 1; in TEST() local
194 TFE_ExecuteOpInNewThread(matmul_op, &retvals[0], &num_retvals, status); in TEST()
346 int num_retvals = 5; in AddEagerOpToGraphAndCheck() local
349 TFE_AddEagerOpToGraph(op, trace_ctx_, retvals, &num_retvals, status_); in AddEagerOpToGraphAndCheck()
354 for (int i = 0; i < num_retvals; ++i) { in AddEagerOpToGraphAndCheck()
373 int num_retvals = 5; in TEST_F() local
375 TFE_AddEagerOpToGraph(op, trace_ctx_, retvals, &num_retvals, status_); in TEST_F()
389 CHECK_EQ(num_retvals, 1); in TEST_F()
|
| D | c_api_experimental.h | 206 TFE_Op* op, TFE_TensorHandle** retvals, int* num_retvals,
301 int* num_retvals, TF_Status* status);
|
| D | c_api_experimental.cc | 8591 int num_retvals = 1; in createTFEQueue() local
8593 TFE_Execute(queueOp.get(), &queue, &num_retvals, status); in createTFEQueue()
8595 CHECK_EQ(num_retvals, 1); in createTFEQueue()
8615 int num_retvals = 0; in createTFEEnqueue() local
8616 TFE_Execute(op, nullptr /*retvals*/, &num_retvals, status); in createTFEEnqueue()
8618 CHECK_EQ(num_retvals, 0); in createTFEEnqueue()
8636 int num_retvals = 1; in createTFEDequeue() local
8637 TFE_Execute(op, &ret, &num_retvals, status); in createTFEDequeue()
8639 CHECK_EQ(num_retvals, 1); in createTFEDequeue()
8780 int* num_retvals, in TFE_ExecuteOpInNewThread() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_xla_computations_pass.cc | 120 const int num_retvals = output_permutation->size(); in RewriteSubgraph() local
125 retvals.reserve(num_retvals); in RewriteSubgraph()
175 for (int i = 0; i < num_retvals; ++i) { in RewriteSubgraph()
177 TF_RETURN_IF_ERROR(GetIndexAttr(*retvals[i], num_retvals, &index)); in RewriteSubgraph()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 341 int num_retvals = node_data->NumOutputs(); in ExecuteFlexOp() local
345 &num_retvals), in ExecuteFlexOp()
348 if (num_retvals != node_data->NumOutputs()) { in ExecuteFlexOp()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 263 virtual size_t num_retvals() const = 0;
289 size_t num_retvals() const override { return ret_types_.size(); } in num_retvals() function
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 2237 int num_retvals = 1; local
2239 TFE_Execute(op, &output_handle, &num_retvals, status);
2749 int num_retvals = 0; local
2763 num_retvals += delta;
2766 tensorflow::gtl::InlinedVector<TFE_TensorHandle*, 2> retvals(num_retvals);
2769 TFE_Execute(op, retvals.data(), &num_retvals, status);
2785 tensorflow::Safe_PyObjectPtr flat_result(PyList_New(num_retvals));
2786 for (int i = 0; i < num_retvals; ++i) {
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 176 size_t num_retvals() const override { in num_retvals() function in tensorflow::data::__anonf5d3769a0111::MapDefunOp::MapFunctionCallFrame
|
| D | captured_function.cc | 183 size_t num_retvals() const override { return retvals_.size(); } in num_retvals() function in tensorflow::data::__anonf8786f2a0211::CallFrameBase
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session.cc | 1753 size_t num_retvals() const override { in num_retvals() function in tensorflow::DirectSession::RunCallableCallFrame
|