Home
last modified time | relevance | path

Searched refs:retvals (Results 1 – 25 of 62) sorted by relevance

123

/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_execute_node.cc30 const gtl::InlinedVector<TensorHandle*, 2>& retvals = retvals_; in RunAsync() local
61 for (size_t i = 0; i < retvals.size(); ++i) { in RunAsync()
62 retvals[i]->PoisonRemote(s, device, context_view_id_); in RunAsync()
72 for (auto handle : retvals) { in RunAsync()
78 [inputs, retvals, call_opts, response, device, in RunAsync()
93 for (size_t i = 0; i < retvals.size(); ++i) { in RunAsync()
99 Status s = retvals[i]->SetRemoteShapeAndDevice( in RunAsync()
106 << retvals[i] in RunAsync()
113 retvals[i]->PoisonRemote(status, device, context_view_id); in RunAsync()
115 retvals[i]->Unref(); in RunAsync()
Deager_service_impl.cc174 const std::vector<int32>& output_nums, TensorHandle** retvals, in AddOpRetvalsToResponse() argument
182 TF_RETURN_IF_ERROR(TensorHandleProto(retvals[i], add_tensor_proto_fn())); in AddOpRetvalsToResponse()
183 retvals[i]->Unref(); in AddOpRetvalsToResponse()
187 TF_RETURN_IF_ERROR(TensorHandleShape(retvals[i], add_shape_proto_fn())); in AddOpRetvalsToResponse()
189 Device* device = retvals[i]->device(); in AddOpRetvalsToResponse()
192 if (retvals[i]->Type() == TensorHandle::REMOTE) { in AddOpRetvalsToResponse()
193 retvals[i]->Unref(); in AddOpRetvalsToResponse()
196 eager_context->RemoteMgr()->AddOperationOutput(retvals[i], op_id, in AddOpRetvalsToResponse()
471 auto* retvals = new absl::FixedArray<TensorHandle*>(*num_retvals); in RunComponentFunction() local
485 op, retvals->data(), num_retvals, in RunComponentFunction()
[all …]
Dremote_execute_node.h46 absl::Span<TensorHandle*> retvals) in RemoteExecuteNode() argument
59 for (auto handle : retvals) { in RemoteExecuteNode()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc384 EagerOperation* op, TensorHandle** retvals, int* num_retvals, in GetOrCreateKernelAndDevice() argument
610 EagerOperation* op, TensorHandle** retvals) { in AddOrExecuteNode() argument
636 retvals[i] = TensorHandle::CreateEmptyLocalHandle( in AddOrExecuteNode()
643 remote_func_params, &ctx, &retvals[i])); in AddOrExecuteNode()
651 absl::Span<TensorHandle*>(retvals, num_outputs), op->GetStackTrace()); in AddOrExecuteNode()
663 retvals[i] = nullptr; in AddOrExecuteNode()
669 {retvals, static_cast<size_t>(num_outputs)}, in AddOrExecuteNode()
694 Status EagerLocalExecute(EagerOperation* op, TensorHandle** 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()
[all …]
Dexecute.h45 Status EagerExecute(EagerOperation* op, TensorHandle** retvals,
55 absl::Span<TensorHandle*> retvals,
85 void EagerLocalExecuteAsync(EagerOperation* op, TensorHandle** retvals,
Dshape_inference.cc31 const gtl::InlinedVector<TensorHandle*, 2>& retvals) { in RunShapeInference() argument
49 CHECK_EQ(ic.num_outputs(), retvals.size()); in RunShapeInference()
52 retvals[i]->SetInferenceShape(&ic, shape_handle); in RunShapeInference()
Dexecute_node.h100 absl::Span<TensorHandle*> retvals, in ExecuteNode() argument
109 retvals_(retvals), in ExecuteNode()
156 absl::Span<TensorHandle*> retvals, in AsyncExecuteNode() argument
168 for (auto handle : retvals) { in AsyncExecuteNode()
Dcustom_device_op_handler.cc55 ImmediateExecutionTensorHandle** retvals, in Execute() argument
62 return custom_device->Execute(op, retvals, num_retvals); in Execute()
93 reinterpret_cast<tensorflow::AbstractTensorHandle**>(retvals), in Execute()
Dshape_inference.h31 const gtl::InlinedVector<TensorHandle*, 2>& retvals);
/external/linux-kselftest/tools/testing/selftests/bpf/verifier/
Djmp32.c19 .retvals = {
51 .retvals = {
90 .retvals = {
114 .retvals = {
153 .retvals = {
177 .retvals = {
216 .retvals = {
243 .retvals = {
284 .retvals = {
311 .retvals = {
[all …]
/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc87 TFE_TensorHandle* retvals[1]; in BM_Execute() local
97 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in BM_Execute()
128 TFE_TensorHandle* retvals[1]; in BM_Execute_Identity() local
136 TFE_Execute(identity, &retvals[0], &num_retvals, status); in BM_Execute_Identity()
440 TFE_TensorHandle* retvals[1]; in TensorHandleSilentCopy() local
442 TFE_Execute(matmul, &retvals[0], &num_retvals, status.get()); in TensorHandleSilentCopy()
449 TFE_DeleteTensorHandle(retvals[0]); in TensorHandleSilentCopy()
576 TFE_TensorHandle* retvals[1]; in TEST() local
578 TFE_Execute(shape_op, &retvals[0], &num_retvals, status.get()); in TEST()
582 device_name = TFE_TensorHandleDeviceName(retvals[0], status.get()); in TEST()
[all …]
Dc_api_distributed_test.cc205 TFE_TensorHandle* retvals[1] = {nullptr}; in TestFunctionWithPackedInput() local
207 TFE_Execute(func, &retvals[0], &num_retvals, status); in TestFunctionWithPackedInput()
212 TF_Tensor* t = TFE_TensorHandleResolve(retvals[0], status); in TestFunctionWithPackedInput()
214 TFE_DeleteTensorHandle(retvals[0]); in TestFunctionWithPackedInput()
391 TFE_TensorHandle* retvals[1] = {nullptr}; in TEST() local
393 TFE_Execute(func, &retvals[0], &num_retvals, status); in TEST()
396 TF_Tensor* t = TFE_TensorHandleResolve(retvals[0], status); in TEST()
398 TFE_DeleteTensorHandle(retvals[0]); in TEST()
515 TFE_TensorHandle* retvals[1] = {nullptr}; in TestDistributedFunctionCancellation() local
517 TFE_Execute(func, &retvals[0], &num_retvals, status); in TestDistributedFunctionCancellation()
[all …]
Dc_api_remote_test.cc81 TFE_TensorHandle* retvals[1]; in TestRemoteExecute() local
83 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in TestRemoteExecute()
86 TF_Tensor* t = TFE_TensorHandleResolve(retvals[0], status); in TestRemoteExecute()
101 TFE_DeleteTensorHandle(retvals[0]); in TestRemoteExecute()
Dc_api_remote_test_util.cc170 TFE_TensorHandle* retvals[1]; in TestRemoteExecuteSilentCopies() local
172 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in TestRemoteExecuteSilentCopies()
185 TFE_TensorHandleBackingDeviceName(retvals[0], status); in TestRemoteExecuteSilentCopies()
191 retvals[0], ctx, "/job:localhost/replica:0/task:0/device:CPU:0", status); in TestRemoteExecuteSilentCopies()
212 TFE_DeleteTensorHandle(retvals[0]); in TestRemoteExecuteSilentCopies()
Dc_api_experimental_test.cc332 TFE_TensorHandle* retvals[2] = {nullptr, nullptr}; in Executor_MatMul_CPU() local
334 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in Executor_MatMul_CPU()
341 TF_Tensor* t = TFE_TensorHandleResolve(retvals[0], status); in Executor_MatMul_CPU()
343 TFE_DeleteTensorHandle(retvals[0]); in Executor_MatMul_CPU()
458 TFE_TensorHandle* retvals[1]; in TEST() local
460 TFE_Execute(shape_op, &retvals[0], &num_retvals, status.get()); in TEST()
463 device_type = TFE_TensorHandleDeviceType(retvals[0], status.get()); in TEST()
467 device_id = TFE_TensorHandleDeviceID(retvals[0], status.get()); in TEST()
472 TFE_DeleteTensorHandle(retvals[0]); in TEST()
Dunified_api_testutil.cc102 int retvals = outputs.size() - null_indices.size(); in RunModel() local
103 std::vector<AbstractTensorHandle*> fn_outputs(retvals); in RunModel()
106 &retvals)); in RunModel()
/external/tensorflow/tensorflow/c/experimental/op_handler/
Dinternal_test.cc30 absl::Span<AbstractTensorHandle*> retvals, in Execute() argument
35 return operation->Execute(retvals, num_retvals); in Execute()
70 std::vector<AbstractTensorHandle*> retvals; 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.cc68 Status OpHandlerOperation::Execute(absl::Span<AbstractTensorHandle*> retvals, in Execute() argument
71 return WrapperOperation::Execute(retvals, num_retvals); in Execute()
73 return handler_->Execute(this, retvals, num_retvals); in Execute()
Dinternal.h43 absl::Span<AbstractTensorHandle*> retvals,
71 Status Execute(absl::Span<AbstractTensorHandle*> retvals,
Dwrapper_operation.cc115 Status WrapperOperation::Execute(absl::Span<AbstractTensorHandle*> retvals, in Execute() argument
117 return parent_op_->Execute(retvals, num_retvals); in Execute()
/external/tensorflow/tensorflow/python/debug/wrappers/
Dframework.py482 retvals, run_end_req = self._run_with_debugging(
486 retvals, run_end_req = self._run_with_profiling(
492 retvals = callable_runner(*callable_runner_args)
498 retvals = callable_object(*callable_runner_args)
500 retvals = self._sess.run(
517 return retvals
560 retvals = callable_runner(*callable_runner_args,
574 retvals = callable_object(
577 retvals = self._sess.run(fetches,
585 retvals = op_error
[all …]
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor_test.cc155 std::vector<Tensor> retvals; in TEST_F() local
156 TF_ASSERT_OK(call_frame.ConsumeRetvals(&retvals, false)); in TEST_F()
157 EXPECT_EQ(3.0, V(retvals[0])); // out = 1.0 + 2.0 = 3.0 in TEST_F()
191 std::vector<Tensor> retvals; in TEST_F() local
192 TF_ASSERT_OK(call_frame.ConsumeRetvals(&retvals, false)); in TEST_F()
193 EXPECT_EQ(1024.0, V(retvals[0])); // b=v10=2*v9=4*v8=...=1024*a=1024.0 in TEST_F()
239 std::vector<Tensor> retvals; in TEST_F() local
240 TF_ASSERT_OK(call_frame.ConsumeRetvals(&retvals, false)); in TEST_F()
241 EXPECT_EQ(4096.0, V(retvals[0])); in TEST_F()
271 std::vector<Tensor> retvals; in TEST_F() local
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass.cc119 std::vector<Node*> retvals; in RewriteSubgraph() local
121 retvals.reserve(num_retvals); in RewriteSubgraph()
131 retvals.push_back(n); in RewriteSubgraph()
151 std::sort(retvals.begin(), retvals.end(), in RewriteSubgraph()
173 TF_RETURN_IF_ERROR(GetIndexAttr(*retvals[i], num_retvals, &index)); in RewriteSubgraph()
175 retvals[i]->AddAttr("index", i); in RewriteSubgraph()
/external/antlr/tool/src/main/java/org/antlr/tool/
DLeftRecursiveRuleAnalyzer.java26 public GrammarAST retvals; field in LeftRecursiveRuleAnalyzer
57 retvals = t; in setReturnValues()
198 ruleST.add("userRetvals", retvals); in getArtificialPrecStartRule()
218 ruleST.add("userRetvals", retvals); in getArtificialOpPrecRule()
248 ruleST.add("userRetvals", retvals); in getArtificialPrimaryRule()
322 if ( retvals==null ) return; in fillRetValAssignments()
325 for (String name : getNamesFromArgAction(retvals.token)) { in fillRetValAssignments()
/external/tensorflow/tensorflow/c/experimental/gradients/tape/
Dtape_operation.cc197 Status TapeOperation::Execute(absl::Span<AbstractTensorHandle*> retvals, in Execute() argument
199 TF_RETURN_IF_ERROR(parent_op_->Execute(retvals, num_retvals)); in Execute()
202 forward_op_.outputs.push_back(retvals[i]); in Execute()

123