Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/c/experimental/ops/
Dmath_ops.cc36 int num_retvals = 1; in Mul() local
37 return mul_op->Execute(outputs, &num_retvals); in Mul()
54 int num_retvals = 1; in Conj() local
55 TF_RETURN_IF_ERROR(conj_op->Execute(outputs, &num_retvals)); in Conj()
71 int num_retvals = 1; in Add() local
72 TF_RETURN_IF_ERROR(add_op->Execute(outputs, &num_retvals)); in Add()
84 int num_retvals = 1; in Sub() local
85 TF_RETURN_IF_ERROR(sub_op->Execute(outputs, &num_retvals)); in Sub()
102 int num_retvals = 1; in MatMul() local
103 TF_RETURN_IF_ERROR(matmul_op->Execute(outputs, &num_retvals)); in MatMul()
[all …]
Darray_ops.cc33 int num_retvals = 1; in Identity() local
34 return identity_op->Execute(outputs, &num_retvals); in Identity()
45 int num_retvals = inputs.size(); in IdentityN() local
46 return identity_n_op->Execute(outputs, &num_retvals); in IdentityN()
56 int num_retvals = 1; in ZerosLike() local
57 return z_op->Execute(outputs, &num_retvals); in ZerosLike()
67 int num_retvals = 1; in Shape() local
68 TF_RETURN_IF_ERROR(shape_op->Execute(outputs, &num_retvals)); in Shape()
80 int num_retvals = 1; in ExpandDims() local
81 return op->Execute(outputs, &num_retvals); in ExpandDims()
[all …]
Dnn_ops.cc38 int num_retvals = 2; in SparseSoftmaxCrossEntropyWithLogits() local
39 TF_RETURN_IF_ERROR(sm_loss_op->Execute(outputs, &num_retvals)); in SparseSoftmaxCrossEntropyWithLogits()
54 int num_retvals = 1; in ReluGrad() local
55 TF_RETURN_IF_ERROR(relugrad_op->Execute(outputs, &num_retvals)); in ReluGrad()
67 int num_retvals = 1; in Relu() local
68 TF_RETURN_IF_ERROR(relu_op->Execute(outputs, &num_retvals)); in Relu()
82 int num_retvals = 1; in BiasAdd() local
83 TF_RETURN_IF_ERROR(bias_add_op->Execute(outputs, &num_retvals)); in BiasAdd()
100 int num_retvals = 1; in BiasAddGrad() local
101 TF_RETURN_IF_ERROR(bias_add_grad_op->Execute(outputs, &num_retvals)); in BiasAddGrad()
/external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/
Dvariable_ops.cc58 int num_retvals = 1; in CreateUninitializedResourceVariable() local
60 absl::MakeSpan(&var_handle, num_retvals), &num_retvals)); in CreateUninitializedResourceVariable()
80 int num_retvals = 0; in AssignVariable() local
81 TF_RETURN_IF_ERROR(assign_op->Execute({}, &num_retvals)); in AssignVariable()
94 int num_retvals = 1; in ReadVariable() local
96 read_op->Execute(absl::MakeSpan(&value, num_retvals), &num_retvals)); in ReadVariable()
113 int num_retvals = 0; in DestroyResource() local
114 TF_RETURN_IF_ERROR(destroy_op->Execute({}, &num_retvals)); in DestroyResource()
Drestore_ops.cc97 int num_retvals = 1; in SingleRestore() local
99 absl::MakeSpan(&restored_handle, num_retvals), &num_retvals)); in SingleRestore()
/external/tensorflow/tensorflow/c/eager/
Dcustom_device_test.cc54 int num_retvals = 1; in TEST() local
55 TFE_Execute(matmul.get(), &retval, &num_retvals, status.get()); in TEST()
123 int num_retvals = 1; in TEST() local
125 TFE_Execute(op.get(), &var_handle, &num_retvals, status.get()); in TEST()
141 num_retvals = 0; in TEST()
142 TFE_Execute(op.get(), nullptr, &num_retvals, status.get()); in TEST()
153 num_retvals = 1; in TEST()
155 TFE_Execute(op.get(), &var_value, &num_retvals, status.get()); in TEST()
177 num_retvals = 0; in TEST()
178 TFE_Execute(op.get(), nullptr, &num_retvals, status.get()); in TEST()
[all …]
Dc_api_test.cc88 int num_retvals = 1; in BM_Execute() local
97 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in BM_Execute()
129 int num_retvals = 1; in BM_Execute_Identity() local
136 TFE_Execute(identity, &retvals[0], &num_retvals, status); in BM_Execute_Identity()
441 int num_retvals = 1; in TensorHandleSilentCopy() local
442 TFE_Execute(matmul, &retvals[0], &num_retvals, status.get()); in TensorHandleSilentCopy()
577 int num_retvals = 1; in TEST() local
578 TFE_Execute(shape_op, &retvals[0], &num_retvals, status.get()); in TEST()
643 int num_retvals = 1; in ExecuteAdd() local
645 TFE_Execute(add_op, &retval, &num_retvals, status); in ExecuteAdd()
[all …]
Dc_api_distributed_test.cc206 int num_retvals = 1; in TestFunctionWithPackedInput() local
207 TFE_Execute(func, &retvals[0], &num_retvals, status); in TestFunctionWithPackedInput()
209 ASSERT_EQ(1, num_retvals); in TestFunctionWithPackedInput()
392 int num_retvals = 1; in TEST() local
393 TFE_Execute(func, &retvals[0], &num_retvals, status); in TEST()
395 ASSERT_EQ(1, num_retvals); in TEST()
516 int num_retvals = 1; in TestDistributedFunctionCancellation() local
517 TFE_Execute(func, &retvals[0], &num_retvals, status); in TestDistributedFunctionCancellation()
523 ASSERT_EQ(1, num_retvals); in TestDistributedFunctionCancellation()
598 int num_retvals = 1; in TestRemoteExecuteDeleteContextWithOutstandingRPC() local
[all …]
Dc_api_cluster_test.cc68 int num_retvals = 1; in CheckRemoteMatMulExecutesOK() local
69 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in CheckRemoteMatMulExecutesOK()
100 int num_retvals = 1; in ReadVariable() local
101 TFE_Execute(op, out_value, &num_retvals, status); in ReadVariable()
309 int num_retvals = 0; in TestRemoteExecuteUpdateServerDefResourceAccess() local
310 TFE_Execute(op, nullptr, &num_retvals, status); in TestRemoteExecuteUpdateServerDefResourceAccess()
322 num_retvals = 0; in TestRemoteExecuteUpdateServerDefResourceAccess()
323 TFE_Execute(op, nullptr, &num_retvals, status); in TestRemoteExecuteUpdateServerDefResourceAccess()
Dc_api_experimental_test.cc294 int num_retvals = 1; in TEST() local
295 TFE_Execute(op, result.data(), &num_retvals, status); in TEST()
298 ASSERT_EQ(num_retvals, 1); in TEST()
333 int num_retvals = 2; in Executor_MatMul_CPU() local
334 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in Executor_MatMul_CPU()
335 EXPECT_EQ(1, num_retvals); in Executor_MatMul_CPU()
459 int num_retvals = 1; in TEST() local
460 TFE_Execute(shape_op, &retvals[0], &num_retvals, status.get()); in TEST()
Dc_api_test_util.cc210 int num_retvals = 1; in TestVariable() local
211 TFE_Execute(op, &var_handle, &num_retvals, status); in TestVariable()
215 CHECK_EQ(1, num_retvals); in TestVariable()
236 num_retvals = 0; in TestVariable()
237 TFE_Execute(op, nullptr, &num_retvals, status); in TestVariable()
240 CHECK_EQ(0, num_retvals); in TestVariable()
Dc_api_remote_test.cc82 int num_retvals = 1; in TestRemoteExecute() local
83 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in TestRemoteExecute()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.cc60 int* num_retvals) { in GetNumRetvals() argument
78 *num_retvals += iter->second.i(); in GetNumRetvals()
86 *num_retvals += iter->second.list().type_size(); in GetNumRetvals()
88 *num_retvals += 1; in GetNumRetvals()
99 int* num_retvals) { in GetEagerOperationAndNumRetvals() argument
145 num_retvals); in GetEagerOperationAndNumRetvals()
173 EagerContext* eager_context, int op_id, int num_retvals, in AddOpRetvalsToResponse() argument
181 for (int i = 0; i < num_retvals; i++) { in AddOpRetvalsToResponse()
186 for (int i = 0; i < num_retvals; i++) { in AddOpRetvalsToResponse()
452 int* num_retvals = new int(0); in RunComponentFunction() local
[all …]
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_testlib.cc54 int num_retvals = 1; in Create() local
55 TFE_Execute(op.get(), &var_handle, &num_retvals, status); in Create()
71 int num_retvals = 0; in Destroy() local
72 TFE_Execute(op.get(), nullptr, &num_retvals, status); in Destroy()
89 int num_retvals = 1; in Read() local
91 TFE_Execute(op.get(), &var_value, &num_retvals, status); in Read()
110 int num_retvals = 0; in GeneralAssignment() local
111 TFE_Execute(op.get(), nullptr, &num_retvals, status); in GeneralAssignment()
174 int num_retvals = num_replicas; in ExtractPerDeviceValues() local
175 TFE_Execute(op.get(), result_handles, &num_retvals, status); in ExtractPerDeviceValues()
[all …]
/external/tensorflow/tensorflow/c/experimental/op_handler/
Dinternal_test.cc31 int* num_retvals) override { in Execute() argument
35 return operation->Execute(retvals, num_retvals); in Execute()
71 int num_retvals = 0; in TEST() local
73 s = op->Execute(absl::Span<AbstractTensorHandle*>(retvals), &num_retvals); in TEST()
88 s = op->Execute(absl::Span<AbstractTensorHandle*>(retvals), &num_retvals); in TEST()
98 s = op->Execute(absl::Span<AbstractTensorHandle*>(retvals), &num_retvals); in TEST()
Dinternal.cc69 int* num_retvals) { in Execute() argument
71 return WrapperOperation::Execute(retvals, num_retvals); in Execute()
73 return handler_->Execute(this, retvals, num_retvals); in Execute()
Dinternal.h44 int* num_retvals) = 0;
72 int* num_retvals) override;
/external/tensorflow/tensorflow/c/experimental/saved_model/internal/
Dsaved_model_api_test.cc130 int num_retvals = 1; in TEST_P() local
132 TFE_Execute(compute_fn_op, &compute_fn_outputs[0], &num_retvals, status); in TEST_P()
269 int num_retvals = TF_SignatureDefParamListSize(returns); in TEST_P() local
271 TFE_Execute(serving_default_op, compute_fn_outputs.data(), &num_retvals, in TEST_P()
326 int num_retvals = 1; in TEST_P() local
328 TFE_Execute(read_file_op, &read_file_fn_outputs[0], &num_retvals, status); in TEST_P()
395 int num_retvals = 1; in TEST_P() local
397 TFE_Execute(lookup_op, &lookup_fn_outputs[0], &num_retvals, status); in TEST_P()
427 int num_retvals = 1; in TEST_P() local
429 TFE_Execute(lookup_op, &lookup_fn_outputs[0], &num_retvals, status); in TEST_P()
[all …]
/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
Drestored_resource.cc37 int num_retvals = 1; in ExecuteNoArgDummyReturnFunction() local
39 absl::MakeSpan(&dummy_output, num_retvals), &num_retvals)); in ExecuteNoArgDummyReturnFunction()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcustom_device_op_handler.cc56 int* num_retvals) { in Execute() argument
62 return custom_device->Execute(op, retvals, num_retvals); in Execute()
94 *num_retvals), in Execute()
95 num_retvals); in Execute()
Dexecute.cc384 EagerOperation* op, TensorHandle** retvals, int* num_retvals, in GetOrCreateKernelAndDevice() argument
562 if (num_outputs > *num_retvals) { in GetOrCreateKernelAndDevice()
565 *num_retvals); in GetOrCreateKernelAndDevice()
567 *num_retvals = num_outputs; in GetOrCreateKernelAndDevice()
695 int* num_retvals) { in EagerLocalExecute() argument
708 auto status = GetOrCreateKernelAndDevice(op, retvals, num_retvals, &kernel); in EagerLocalExecute()
721 status = GetOrCreateKernelAndDevice(op, retvals, num_retvals, &kernel); in EagerLocalExecute()
774 int num_retvals = 1; in MaybePackInputTensor() local
775 absl::FixedArray<tensorflow::TensorHandle*> retvals(num_retvals); in MaybePackInputTensor()
777 EagerLocalExecute(&pack_op, retvals.data(), &num_retvals)); in MaybePackInputTensor()
[all …]
Dexecute.h46 int* num_retvals);
86 int* num_retvals, StatusCallback done);
/external/tensorflow/tensorflow/java/src/main/native/
Deager_operation_builder_jni.cc95 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/c/experimental/gradients/tape/
Dtape_operation.cc198 int* num_retvals) { in Execute() argument
199 TF_RETURN_IF_ERROR(parent_op_->Execute(retvals, num_retvals)); in Execute()
200 for (int i = 0; i < *num_retvals; i++) { in Execute()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass.cc116 const int num_retvals = output_permutation->size(); in RewriteSubgraph() local
121 retvals.reserve(num_retvals); in RewriteSubgraph()
171 for (int i = 0; i < num_retvals; ++i) { in RewriteSubgraph()
173 TF_RETURN_IF_ERROR(GetIndexAttr(*retvals[i], num_retvals, &index)); in RewriteSubgraph()

123