Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc72 TFE_TensorHandle* retvals[1]; in BM_Execute() local
76 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in BM_Execute()
174 TFE_TensorHandle* retvals[1]; in TestRemoteExecute() local
176 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in TestRemoteExecute()
179 TF_Tensor* t = TFE_TensorHandleResolve(retvals[0], status); in TestRemoteExecute()
194 TFE_DeleteTensorHandle(retvals[0]); in TestRemoteExecute()
256 TFE_TensorHandle* retvals[1]; in TestRemoteExecuteSilentCopies() local
258 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in TestRemoteExecuteSilentCopies()
262 retvals[0], ctx, "/job:localhost/replica:0/task:0/device:CPU:0", status); in TestRemoteExecuteSilentCopies()
280 TFE_DeleteTensorHandle(retvals[0]); in TestRemoteExecuteSilentCopies()
[all …]
Dc_api_experimental_test.cc53 TFE_TensorHandle* retvals[1] = {nullptr}; in ExecuteWithProfiling() local
65 TFE_Execute(matmul, &retvals[0], &num_retvals, status); in ExecuteWithProfiling()
89 TF_Tensor* t = TFE_TensorHandleResolve(retvals[0], status); in ExecuteWithProfiling()
90 TFE_DeleteTensorHandle(retvals[0]); in ExecuteWithProfiling()
Dc_api.h386 TF_CAPI_EXPORT extern void TFE_Execute(TFE_Op* op, TFE_TensorHandle** retvals,
Dc_api.cc810 void TFE_Execute(TFE_Op* op, TFE_TensorHandle** retvals, int* num_retvals, in TFE_Execute() argument
821 retvals[i] = new TFE_TensorHandle(handle_retvals[i]); in TFE_Execute()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc367 gtl::InlinedVector<TensorHandle*, 2>* retvals, in EagerLocalExecute() argument
498 retvals->resize(*num_retvals); in EagerLocalExecute()
506 (*retvals)[i] = new TensorHandle( in EagerLocalExecute()
514 graph_collector, output_dtypes, *retvals); in EagerLocalExecute()
521 retvals->data(), *num_retvals); in EagerLocalExecute()
640 Status EagerRemoteExecute(EagerOperation* op, TensorHandle** retvals, in EagerRemoteExecute() argument
728 retvals[i] = new TensorHandle( in EagerRemoteExecute()
742 retvals_copy.push_back(retvals[i]); in EagerRemoteExecute()
749 [](const gtl::InlinedVector<TensorHandle*, 2>& retvals, in EagerRemoteExecute()
752 for (int i = 0; i < retvals.size(); i++) { in EagerRemoteExecute()
[all …]
Dexecute.h41 tensorflow::gtl::InlinedVector<tensorflow::TensorHandle*, 2>* retvals,
51 TensorHandle** retvals, int num_retvals);
Dexecute_node.h39 const tensorflow::gtl::InlinedVector<TensorHandle*, 2>& retvals) in ExecuteNode() argument
47 retvals_(retvals) { in ExecuteNode()
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor_test.cc152 std::vector<Tensor> retvals; in TEST_F() local
153 TF_ASSERT_OK(call_frame.ConsumeRetvals(&retvals, false)); in TEST_F()
154 EXPECT_EQ(2.0, V(retvals[0])); // out = 1.0 + 1.0 = 2.0 in TEST_F()
180 std::vector<Tensor> retvals; in TEST_F() local
181 TF_ASSERT_OK(call_frame.ConsumeRetvals(&retvals, false)); in TEST_F()
182 EXPECT_EQ(1024.0, V(retvals[0])); // b=v10=2*v9=4*v8=...=1024*a=1024.0 in TEST_F()
228 std::vector<Tensor> retvals; in TEST_F() local
229 TF_ASSERT_OK(call_frame.ConsumeRetvals(&retvals, false)); in TEST_F()
230 EXPECT_EQ(4096.0, V(retvals[0])); in TEST_F()
Dcaptured_function.cc170 Status ConsumeRetvals(std::vector<Tensor>* retvals) { in ConsumeRetvals() argument
171 retvals->reserve(retvals_.size()); in ConsumeRetvals()
177 retvals->emplace_back(std::move(val.value())); in ConsumeRetvals()
/external/tensorflow/tensorflow/c/
Dc_api_experimental_test.cc190 TFE_TensorHandle* retvals[1] = {nullptr}; in TEST() local
194 TFE_ExecuteOpInNewThread(matmul_op, &retvals[0], &num_retvals, status); in TEST()
199 TF_Tensor* t = TFE_TensorHandleResolve(retvals[0], status); in TEST()
213 TFE_DeleteTensorHandle(retvals[0]); in TEST()
345 TFE_TensorHandle* retvals[5]; in AddEagerOpToGraphAndCheck() local
349 TFE_AddEagerOpToGraph(op, trace_ctx_, retvals, &num_retvals, status_); in AddEagerOpToGraphAndCheck()
355 TFE_DeleteTensorHandle(retvals[i]); in AddEagerOpToGraphAndCheck()
372 TFE_TensorHandle* retvals[5]; in TEST_F() local
375 TFE_AddEagerOpToGraph(op, trace_ctx_, retvals, &num_retvals, status_); in TEST_F()
390 CHECK_EQ(TFE_TensorHandleDataType(retvals[0]), TF_FLOAT); in TEST_F()
[all …]
Dc_api_experimental.h206 TFE_Op* op, TFE_TensorHandle** retvals, int* num_retvals,
300 TFE_Op* op, TFE_TraceContext* trace_ctx, TFE_TensorHandle** retvals,
Dc_api_experimental.cc8779 TFE_TensorHandle** retvals, in TFE_ExecuteOpInNewThread() argument
8786 [op, retvals, num_retvals, n]() { in TFE_ExecuteOpInNewThread()
8787 TFE_Execute(op, retvals, num_retvals, n->status.get()); in TFE_ExecuteOpInNewThread()
9057 TFE_TensorHandle** retvals, in TFE_AddEagerOpToGraph() argument
9112 retvals[i] = TFE_NewTensorHandleFromTFOutput(output, dtype); in TFE_AddEagerOpToGraph()
/external/tensorflow/tensorflow/python/debug/wrappers/
Dframework.py521 retvals = callable_runner(*callable_runner_args,
535 retvals = callable_object(
538 retvals = self._sess.run(fetches,
546 retvals = op_error
559 retvals = callable_runner(*callable_runner_args,
563 retvals = self._sess.run(fetches,
581 retvals = self.invoke_node_stepper(
585 retvals = self._sess.run(
602 return retvals
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass.cc123 std::vector<Node*> retvals; in RewriteSubgraph() local
125 retvals.reserve(num_retvals); in RewriteSubgraph()
135 retvals.push_back(n); in RewriteSubgraph()
155 std::sort(retvals.begin(), retvals.end(), in RewriteSubgraph()
177 TF_RETURN_IF_ERROR(GetIndexAttr(*retvals[i], num_retvals, &index)); in RewriteSubgraph()
179 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/core/distributed_runtime/eager/
Deager_service_impl.cc190 tensorflow::gtl::InlinedVector<tensorflow::TensorHandle*, 2> retvals; in ExecuteOp() local
191 TF_RETURN_IF_ERROR(EagerExecute(op.get(), &retvals, &num_retvals)); in ExecuteOp()
193 server_context->AddOperationOutputs(retvals, operation.id()); in ExecuteOp()
195 for (auto* handle : retvals) { in ExecuteOp()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc159 const std::vector<XlaExpression>& retvals, in BuildComputation() argument
186 elems.reserve(retvals.size()); in BuildComputation()
192 for (int i = 0; i < retvals.size(); ++i) { in BuildComputation()
194 const XlaExpression& retval = retvals[i]; in BuildComputation()
1144 result->outputs.resize(context->retvals().size()); in CompileGraph()
1145 std::vector<XlaExpression> retvals = context->retvals(); in CompileGraph() local
1148 client(), absl::Span<XlaExpression>(retvals)); in CompileGraph()
1154 absl::Span<XlaExpression>(retvals)); in CompileGraph()
1159 ConvertConstantsToExpressions(&builder, absl::Span<XlaExpression>(retvals)); in CompileGraph()
1162 real_args, retvals, arg_cores, retval_cores, context->resources(), in CompileGraph()
[all …]
Dxla_context.h60 const std::vector<XlaExpression>& retvals() { return retvals_; } in retvals() function
/external/swiftshader/third_party/llvm-7.0/llvm/test/CodeGen/ARM/
Dfast-isel-call-multi-reg-return.ll5 ; Fast-isel can't handle non-double multi-reg retvals.
/external/llvm/test/CodeGen/ARM/
Dfast-isel-call-multi-reg-return.ll5 ; Fast-isel can't handle non-double multi-reg retvals.
/external/swiftshader/third_party/llvm-7.0/llvm/test/Transforms/DeadArgElim/
Daggregates.ll89 ; Case 4: we can remove some retvals from the array. It's nice to produce an
/external/llvm/test/Transforms/DeadArgElim/
Daggregates.ll89 ; Case 4: we can remove some retvals from the array. It's nice to produce an
/external/tensorflow/tensorflow/python/autograph/converters/
Dcontrol_flow.py66 return_stmt = templates.replace(template, retvals=returns)
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_test.cc308 std::vector<Tensor> retvals; in InstantiateAndRunViaCallFrameInterface() local
309 TF_RETURN_IF_ERROR(frame.GetRetvals(&retvals)); in InstantiateAndRunViaCallFrameInterface()
310 CHECK_EQ(rets.size(), retvals.size()); in InstantiateAndRunViaCallFrameInterface()
312 *rets[i] = retvals[i]; in InstantiateAndRunViaCallFrameInterface()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc2766 tensorflow::gtl::InlinedVector<TFE_TensorHandle*, 2> retvals(num_retvals); local
2769 TFE_Execute(op, retvals.data(), &num_retvals, status);
2787 PyList_SET_ITEM(flat_result.get(), i, EagerTensorFromHandle(retvals[i]));