Searched refs:retvals_ (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute_node.h | 109 retvals_(retvals), in ExecuteNode()
125 graph_collector_, cancellation_manager_, retvals_, in Run()
144 absl::Span<TensorHandle*> retvals_; variable
170 retvals_.push_back(handle); in AsyncExecuteNode()
181 for (auto handle : retvals_) { in ~AsyncExecuteNode()
204 cancellation_manager_, absl::MakeSpan(retvals_), stack_trace_); in Run()
220 for (auto handle : retvals_) { in Abort()
240 absl::InlinedVector<TensorHandle*, 2> retvals_; variable
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_execute_node.h | 61 retvals_.push_back(handle); in RemoteExecuteNode()
83 for (auto handle : retvals_) { in ~RemoteExecuteNode()
94 return RunShapeInference(ndef_, *lib_def_, inputs_, retvals_); in Prepare()
103 for (auto handle : retvals_) { in Abort()
135 gtl::InlinedVector<TensorHandle*, 2> retvals_; variable
|
| D | remote_execute_node.cc | 30 const gtl::InlinedVector<TensorHandle*, 2>& retvals = retvals_; in RunAsync()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_context.cc | 73 const int64 retvals_size = retvals_.size(); in SetRetval()
75 retvals_.resize(index + 1); in SetRetval()
77 retvals_[index] = expression; in SetRetval()
|
| D | xla_context.h | 70 const std::vector<XlaExpression>& retvals() { return retvals_; } in retvals()
121 std::vector<XlaExpression> retvals_; variable
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 279 : ret_types_(ret_types), retvals_(ret_types.size()) {} in CallFrameBase()
283 retvals->reserve(retvals_.size()); in ConsumeRetvals()
285 for (auto&& val : retvals_) { in ConsumeRetvals()
295 size_t num_retvals() const override { return retvals_.size(); } in num_retvals()
299 const int retvals_size = retvals_.size(); in SetRetval()
301 !retvals_[index]) { in SetRetval()
302 retvals_[index] = val; in SetRetval()
320 std::vector<gtl::optional<Tensor>> retvals_; member in tensorflow::data::__anon5a9b84710111::CallFrameBase
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | functional_ops.cc | 452 : args_(args), retvals_(retvals), ret_types_(ret_types) {} in BodyFuncCallFrame()
455 size_t num_retvals() const override { return retvals_->size(); } in num_retvals()
479 } else if (TF_PREDICT_FALSE(index >= retvals_->size())) { in SetRetval()
489 (*retvals_)[index] = val; in SetRetval()
495 std::vector<Tensor>* const retvals_; // Not owned. member in tensorflow::__anonf02e412f0111::WhileOp::BodyFuncCallFrame
|