/external/tensorflow/tensorflow/c/experimental/ops/ |
D | math_ops.cc | 36 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 …]
|
D | array_ops.cc | 33 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 …]
|
D | nn_ops.cc | 38 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/ |
D | variable_ops.cc | 58 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()
|
D | restore_ops.cc | 97 int num_retvals = 1; in SingleRestore() local 99 absl::MakeSpan(&restored_handle, num_retvals), &num_retvals)); in SingleRestore()
|
/external/tensorflow/tensorflow/c/eager/ |
D | custom_device_test.cc | 54 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 …]
|
D | c_api_test.cc | 88 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 …]
|
D | c_api_distributed_test.cc | 206 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 …]
|
D | c_api_cluster_test.cc | 68 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()
|
D | c_api_experimental_test.cc | 294 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()
|
D | c_api_test_util.cc | 210 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()
|
D | c_api_remote_test.cc | 82 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/ |
D | eager_service_impl.cc | 60 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/ |
D | parallel_device_testlib.cc | 54 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/ |
D | internal_test.cc | 31 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()
|
D | internal.cc | 69 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()
|
D | internal.h | 44 int* num_retvals) = 0; 72 int* num_retvals) override;
|
/external/tensorflow/tensorflow/c/experimental/saved_model/internal/ |
D | saved_model_api_test.cc | 130 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/ |
D | restored_resource.cc | 37 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/ |
D | custom_device_op_handler.cc | 56 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()
|
D | execute.cc | 384 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 …]
|
D | execute.h | 46 int* num_retvals); 86 int* num_retvals, StatusCallback done);
|
/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/c/experimental/gradients/tape/ |
D | tape_operation.cc | 198 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/ |
D | encapsulate_xla_computations_pass.cc | 116 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()
|