Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/c/experimental/ops/
Dmath_ops.cc45 int num_retvals = 1; in Mul() local
46 return op_ptr->Execute(absl::MakeSpan(z, 1), &num_retvals); in Mul()
74 int num_retvals = 1; in Conj() local
75 return op_ptr->Execute(absl::MakeSpan(output, 1), &num_retvals); in Conj()
93 int num_retvals = 1; in AddV2() local
94 return op_ptr->Execute(absl::MakeSpan(z, 1), &num_retvals); in AddV2()
119 int num_retvals = 1; in MatMul() local
120 return op_ptr->Execute(absl::MakeSpan(product, 1), &num_retvals); in MatMul()
135 int num_retvals = 1; in Neg() local
136 return op_ptr->Execute(absl::MakeSpan(y, 1), &num_retvals); in Neg()
[all …]
Darray_ops.cc43 int num_retvals = 1; in Identity() local
44 return op_ptr->Execute(absl::MakeSpan(output, 1), &num_retvals); in Identity()
75 int num_retvals = output.size(); in IdentityN() local
76 return op_ptr->Execute(output, &num_retvals); in IdentityN()
90 int num_retvals = 1; in ZerosLike() local
91 return op_ptr->Execute(absl::MakeSpan(y, 1), &num_retvals); in ZerosLike()
115 int num_retvals = 1; in Shape() local
116 return op_ptr->Execute(absl::MakeSpan(output, 1), &num_retvals); in Shape()
162 int num_retvals = 1; in ExpandDims() local
163 return op_ptr->Execute(absl::MakeSpan(output, 1), &num_retvals); in ExpandDims()
[all …]
Dnn_ops.cc54 int num_retvals = 2; in SparseSoftmaxCrossEntropyWithLogits() local
56 Status status = op_ptr->Execute(temp_outputs, &num_retvals); in SparseSoftmaxCrossEntropyWithLogits()
75 int num_retvals = 1; in ReluGrad() local
76 return op_ptr->Execute(absl::MakeSpan(backprops, 1), &num_retvals); in ReluGrad()
94 int num_retvals = 1; in Relu() local
95 return op_ptr->Execute(absl::MakeSpan(activations, 1), &num_retvals); in Relu()
115 int num_retvals = 1; in BiasAdd() local
116 return op_ptr->Execute(absl::MakeSpan(output, 1), &num_retvals); in BiasAdd()
136 int num_retvals = 1; in BiasAddGrad() local
137 return op_ptr->Execute(absl::MakeSpan(output, 1), &num_retvals); in BiasAddGrad()
Dresource_variable_ops.cc51 int num_retvals = 1; in VarHandleOp() local
52 return op_ptr->Execute(absl::MakeSpan(resource, 1), &num_retvals); in VarHandleOp()
74 int num_retvals = 1; in ReadVariableOp() local
75 return op_ptr->Execute(absl::MakeSpan(value, 1), &num_retvals); in ReadVariableOp()
94 int num_retvals = 0; in AssignVariableOp() local
96 return op_ptr->Execute(absl::MakeSpan(dummy_outputs), &num_retvals); in AssignVariableOp()
115 int num_retvals = 0; in DestroyResourceOp() local
117 return op_ptr->Execute(absl::MakeSpan(dummy_outputs), &num_retvals); in DestroyResourceOp()
Dio_ops.cc64 int num_retvals = tensors.size(); in RestoreV2() local
65 return op_ptr->Execute(tensors, &num_retvals); in RestoreV2()
87 int num_retvals = 0; in SaveV2() local
89 return op_ptr->Execute(absl::MakeSpan(dummy_outputs), &num_retvals); in SaveV2()
/external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/
Dvariable_ops.cc57 int num_retvals = 1; in CreateUninitializedResourceVariable() local
59 absl::MakeSpan(&var_handle, num_retvals), &num_retvals)); in CreateUninitializedResourceVariable()
79 int num_retvals = 0; in AssignVariable() local
80 TF_RETURN_IF_ERROR(assign_op->Execute({}, &num_retvals)); in AssignVariable()
93 int num_retvals = 1; in ReadVariable() local
95 read_op->Execute(absl::MakeSpan(&value, num_retvals), &num_retvals)); in ReadVariable()
112 int num_retvals = 0; in DestroyResource() local
113 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.cc91 int num_retvals = 1; in BM_Execute() local
99 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in BM_Execute()
129 int num_retvals = 1; in BM_Execute_Identity() local
135 TFE_Execute(identity, &retvals[0], &num_retvals, status); in BM_Execute_Identity()
439 int num_retvals = 1; in TensorHandleSilentCopy() local
440 TFE_Execute(matmul, &retvals[0], &num_retvals, status.get()); in TensorHandleSilentCopy()
575 int num_retvals = 1; in TEST() local
576 TFE_Execute(shape_op, &retvals[0], &num_retvals, status.get()); in TEST()
646 int num_retvals = 1; in ExecuteAdd()
656 TFE_Execute(add_op_dummy, &dummy, &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.cc296 int num_retvals = 1; in TEST() local
297 TFE_Execute(op, result.data(), &num_retvals, status); in TEST()
300 ASSERT_EQ(num_retvals, 1); in TEST()
336 int num_retvals = 2; in Executor_MatMul_CPU() local
337 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in Executor_MatMul_CPU()
338 EXPECT_EQ(1, num_retvals); in Executor_MatMul_CPU()
462 int num_retvals = 1; in TEST() local
463 TFE_Execute(shape_op, &retvals[0], &num_retvals, status.get()); in TEST()
/external/tensorflow/tensorflow/c/experimental/ops/gen/cpp/golden/
Dtesting_ops.cc.golden40 int num_retvals = 1;
41 return op_ptr->Execute(absl::MakeSpan(y, 1), &num_retvals);
56 int num_retvals = 1;
57 return op_ptr->Execute(absl::MakeSpan(product, 1), &num_retvals);
69 int num_retvals = output.size();
70 return op_ptr->Execute(output, &num_retvals);
83 int num_retvals = 2;
85 Status status = op_ptr->Execute(temp_outputs, &num_retvals);
102 int num_retvals = 0;
104 return op_ptr->Execute(absl::MakeSpan(dummy_outputs), &num_retvals);
[all …]
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.cc58 int* num_retvals) { in GetNumRetvals() argument
76 *num_retvals += iter->second.i(); in GetNumRetvals()
84 *num_retvals += iter->second.list().type_size(); in GetNumRetvals()
86 *num_retvals += 1; in GetNumRetvals()
97 int* num_retvals) { in GetEagerOperationAndNumRetvals() argument
145 num_retvals); in GetEagerOperationAndNumRetvals()
173 EagerContext* eager_context, int op_id, int num_retvals, in AddOpRetvalsToResponse() argument
186 for (int i = 0; i < num_retvals; i++) { in AddOpRetvalsToResponse()
191 for (int i = 0; i < num_retvals; i++) { in AddOpRetvalsToResponse()
486 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/core/distributed_runtime/integration_test/
Dc_api_coordination_test.cc157 int num_retvals = 1; in TEST() local
158 TFE_Execute(allreduce, &retvals[0], &num_retvals, status); in TEST()
217 int num_retvals = 0; in TEST() local
218 TFE_Execute(set_op, nullptr, &num_retvals, status); in TEST()
233 num_retvals = 1; in TEST()
234 TFE_Execute(get_op, retvals, &num_retvals, status); in TEST()
382 int num_retvals = 1; in TEST() local
383 TFE_Execute(allreduce, &retvals[0], &num_retvals, status); in TEST()
474 int num_retvals = 1; in TEST_P() local
479 TFE_Execute(get_op, retvals, &num_retvals, status); in TEST_P()
[all …]
Dc_api_multi_client_test.cc101 int num_retvals = 1; in TEST() local
102 TFE_Execute(allreduce, &retvals[0], &num_retvals, status); in TEST()
193 int num_retvals = 1; in TEST() local
194 TFE_Execute(sendop, &retvals[0], &num_retvals, status); in TEST()
205 int num_retvals = 1; in TEST() local
206 TFE_Execute(recvop, &retvals[0], &num_retvals, status); in TEST()
Dc_api_multi_client_function_test.cc284 int num_retvals = 0; in TEST_P() local
287 TFE_Execute(send_func, nullptr, &num_retvals, status); in TEST_P()
326 int num_retvals = 1; in TEST_P() local
329 TFE_Execute(recv_func, &retvals[0], &num_retvals, status); in TEST_P()
/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()
426 int num_retvals = 1; in TEST_P() local
428 TFE_Execute(lookup_op, &lookup_fn_outputs[0], &num_retvals, status); in TEST_P()
[all …]
/external/tensorflow/tensorflow/core/function/
Druntime_client.cc166 int num_retvals = fn_def->signature().output_arg_size(); in CallFunction() local
167 int actual_retvals = num_retvals; in CallFunction()
168 std::vector<ImmediateExecutionTensorHandle*> retvals(num_retvals); in CallFunction()
172 num_retvals), in CallFunction()
175 DCHECK(num_retvals == actual_retvals); in CallFunction()
/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.h46 int* num_retvals);
86 int* num_retvals, StatusCallback done);
Dcontext_test.cc280 int num_retvals = retvals.size(); in TEST_F() local
283 fail_op->Execute(absl::MakeSpan(retvals), &num_retvals)); in TEST_F()
302 TF_ASSERT_OK(success_op->Execute(absl::MakeSpan(retvals), &num_retvals)); in TEST_F()
335 int num_retvals = retvals.size(); in TEST_F() local
336 TF_ASSERT_OK(op->Execute(absl::MakeSpan(retvals), &num_retvals)); in TEST_F()
/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()

123