| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test.cc | 74 int num_retvals = 1; in BM_Execute() local
77 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in BM_Execute()
109 int num_retvals = 1; in BM_Execute_Identity() local
112 TFE_Execute(identity, &retvals[0], &num_retvals, status); in BM_Execute_Identity()
391 int num_retvals = 1; in TensorHandleSilentCopy() local
392 TFE_Execute(matmul, &retvals[0], &num_retvals, status.get()); in TensorHandleSilentCopy()
439 int num_retvals = 1; in TensorHandleSilentCopyLocal() local
440 TFE_Execute(matmul, &retvals[0], &num_retvals, status.get()); in TensorHandleSilentCopyLocal()
564 int num_retvals = 1; in TEST() local
565 TFE_Execute(shape_op, &retvals[0], &num_retvals, status.get()); in TEST()
[all …]
|
| D | c_api_experimental_test.cc | 54 int num_retvals = 1; in ExecuteWithProfiling() local
63 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in ExecuteWithProfiling()
69 ASSERT_EQ(1, num_retvals); in ExecuteWithProfiling()
348 int num_retvals = 1; in TEST() local
349 TFE_Execute(op, result.data(), &num_retvals, status); in TEST()
352 ASSERT_EQ(num_retvals, 1); in TEST()
427 int num_retvals = 1; in TEST() local
428 TFE_Execute(op, result.data(), &num_retvals, status); in TEST()
431 ASSERT_EQ(num_retvals, 1); in TEST()
467 int num_retvals = 2; in Executor_MatMul_CPU() local
[all …]
|
| D | c_api_remote_test.cc | 92 int num_retvals = 1; in TestRemoteExecute() local
93 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in TestRemoteExecute()
176 int num_retvals = 1; in TestRemoteExecuteSilentCopies() local
177 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in TestRemoteExecuteSilentCopies()
265 int num_retvals = 1; in TestRemoteExecuteDeleteContextWithOutstandingRPC() local
266 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in TestRemoteExecuteDeleteContextWithOutstandingRPC()
320 int num_retvals = 1; in CheckRemoteMatMulExecutesOK() local
321 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in CheckRemoteMatMulExecutesOK()
|
| D | c_api.h | 380 int* num_retvals, TF_Status* status);
|
| D | c_api.cc | 1487 void TFE_Execute(TFE_Op* op, TFE_TensorHandle** retvals, int* num_retvals, in TFE_Execute() argument
1489 absl::FixedArray<tensorflow::TensorHandle*> handle_retvals(*num_retvals); in TFE_Execute()
1492 handle_retvals.data(), num_retvals); in TFE_Execute()
1496 for (int i = 0; i < *num_retvals; ++i) { in TFE_Execute()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 57 int* num_retvals) { in GetNumRetvals() argument
75 *num_retvals += iter->second.i(); in GetNumRetvals()
83 *num_retvals += iter->second.list().type_size(); in GetNumRetvals()
85 *num_retvals += 1; in GetNumRetvals()
377 int num_retvals = 0; in ExecuteOp() local
380 operation.attrs(), &num_retvals)); in ExecuteOp()
382 absl::FixedArray<tensorflow::TensorHandle*> retvals(num_retvals); in ExecuteOp()
384 TF_RETURN_IF_ERROR(EagerExecute(op.get(), retvals.data(), &num_retvals)); in ExecuteOp()
387 absl::MakeSpan(retvals.data(), num_retvals), operation.id()); in ExecuteOp()
389 for (int i = 0; i < num_retvals; i++) { in ExecuteOp()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental.cc | 370 int num_retvals = 1; in createTFEQueue() local
372 TFE_Execute(queueOp.get(), &queue, &num_retvals, status); in createTFEQueue()
374 CHECK_EQ(num_retvals, 1); in createTFEQueue()
394 int num_retvals = 0; in createTFEEnqueue() local
395 TFE_Execute(op, nullptr /*retvals*/, &num_retvals, status); in createTFEEnqueue()
397 CHECK_EQ(num_retvals, 0); in createTFEEnqueue()
415 int num_retvals = 1; in createTFEDequeue() local
416 TFE_Execute(op, &ret, &num_retvals, status); in createTFEDequeue()
418 CHECK_EQ(num_retvals, 1); in createTFEDequeue()
549 int* num_retvals, in TFE_ExecuteOpInNewThread() argument
[all …]
|
| D | c_api_experimental_test.cc | 99 int num_retvals = 1; in TEST() local
102 TFE_ExecuteOpInNewThread(matmul_op, &retvals[0], &num_retvals, status); in TEST()
|
| D | c_api_experimental.h | 208 TFE_Op* op, TFE_TensorHandle** retvals, int* num_retvals,
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | eager_operation_builder_jni.cc | 95 int num_retvals = MAX_OUTPUTS_PER_OP; in Java_org_tensorflow_EagerOperationBuilder_execute() local
97 new TFE_TensorHandle*[num_retvals]); in Java_org_tensorflow_EagerOperationBuilder_execute()
99 TFE_Execute(op, retvals.get(), &num_retvals, status); in Java_org_tensorflow_EagerOperationBuilder_execute()
105 jlongArray rethandles = env->NewLongArray(num_retvals); in Java_org_tensorflow_EagerOperationBuilder_execute()
106 if (num_retvals > 0) { in Java_org_tensorflow_EagerOperationBuilder_execute()
108 for (int i = 0; i < num_retvals; ++i) { in Java_org_tensorflow_EagerOperationBuilder_execute()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 348 int* num_retvals) { in EagerLocalExecute() argument
542 if (num_outputs > *num_retvals) { in EagerLocalExecute()
545 *num_retvals); in EagerLocalExecute()
547 *num_retvals = num_outputs; in EagerLocalExecute()
629 int* num_retvals) { in EagerRemoteExecute() argument
721 if (num_outputs != *num_retvals) { in EagerRemoteExecute()
725 *num_retvals = num_outputs; in EagerRemoteExecute()
900 int* num_retvals) { in EagerExecute() argument
920 return EagerLocalExecute(op, retvals, num_retvals); in EagerExecute()
939 return EagerRemoteExecute(op, retvals, num_retvals); in EagerExecute()
|
| D | execute.h | 46 int* num_retvals);
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_xla_computations_pass.cc | 121 const int num_retvals = output_permutation->size(); in RewriteSubgraph() local
126 retvals.reserve(num_retvals); in RewriteSubgraph()
176 for (int i = 0; i < num_retvals; ++i) { in RewriteSubgraph()
178 TF_RETURN_IF_ERROR(GetIndexAttr(*retvals[i], num_retvals, &index)); in RewriteSubgraph()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 332 int num_retvals = node_data->NumOutputs(); in ExecuteFlexOp() local
336 &num_retvals), in ExecuteFlexOp()
339 if (num_retvals != node_data->NumOutputs()) { in ExecuteFlexOp()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 274 virtual size_t num_retvals() const = 0;
300 size_t num_retvals() const override { return ret_types_.size(); } in num_retvals() function
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 1374 rets->reserve(frame->num_retvals()); in Run()
1386 if (rets->size() != frame->num_retvals()) { in Run()
1390 frame->num_retvals(), ").")); in Run()
1394 for (size_t i = 0; i < frame->num_retvals(); ++i) { in Run()
|
| D | direct_session.cc | 1862 size_t num_retvals() const override { in num_retvals() function in tensorflow::DirectSession::RunCallableCallFrame
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 86 size_t num_retvals() const override { in num_retvals() function in tensorflow::data::MapDefunOp::MapFunctionCallFrame
|
| D | captured_function.cc | 271 size_t num_retvals() const override { return retvals_.size(); } in num_retvals() function in tensorflow::data::__anon173dfe950111::CallFrameBase
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 3253 int num_retvals = 1; local
3255 TFE_Execute(op, &output_handle, &num_retvals, status);
3725 int num_retvals = 0; local
3739 num_retvals += delta;
3742 tensorflow::gtl::InlinedVector<TFE_TensorHandle*, 2> retvals(num_retvals);
3745 TFE_Execute(op, retvals.data(), &num_retvals, status);
3761 tensorflow::Safe_PyObjectPtr flat_result(PyList_New(num_retvals));
3762 for (int i = 0; i < num_retvals; ++i) {
|