Searched refs:captured_inputs_ (Results 1 – 2 of 2) sorted by relevance
218 captured_inputs_(captured_inputs) {} in OwnedArgsCallFrame()221 return args_.size() + captured_inputs_->size(); in num_args()231 } else if (index < args_.size() + captured_inputs_->size()) { in GetArg()232 *val = (*captured_inputs_)[index - args_.size()]; in GetArg()234 } else if (index >= args_.size() + captured_inputs_->size()) { in GetArg()244 const std::vector<Tensor>* const captured_inputs_; // Not owned. member in tensorflow::data::__anoncde944130211::OwnedArgsCallFrame254 captured_inputs_(captured_inputs) {} in BorrowedArgsCallFrame()257 return args_.size() + captured_inputs_->size(); in num_args()265 } else if (index < args_.size() + captured_inputs_->size()) { in GetArg()266 *val = (*captured_inputs_)[index - args_.size()]; in GetArg()[all …]
152 const std::vector<Tensor>& captured_inputs() { return captured_inputs_; } in captured_inputs()160 const std::vector<Tensor> captured_inputs_; variable