Searched refs:captured_inputs_ (Results 1 – 3 of 3) sorted by relevance
36 captured_inputs_(std::move(captured_inputs)), in MapDefunOpParams()45 input_tensors.insert(input_tensors.end(), captured_inputs_.begin(), in GetInputTensors()46 captured_inputs_.end()); in GetInputTensors()53 input_names->reserve(arguments_.size() + captured_inputs_.size()); in GetInputNames()58 for (int i = 0; i < captured_inputs_.size(); ++i) { in GetInputNames()82 std::vector<Tensor> captured_inputs_; member in tensorflow::data::__anonfe0e44000111::MapDefunOpParams
331 captured_inputs_(captured_inputs) {} in OwnedArgsCallFrame()334 return args_.size() + captured_inputs_->size(); in num_args()340 const int captured_inputs_size = captured_inputs_->size(); in GetArg()345 *val = &(*captured_inputs_)[index - args_.size()]; in GetArg()365 const std::vector<Tensor>* const captured_inputs_; // Not owned. member in tensorflow::data::__anon5a9b84710111::OwnedArgsCallFrame375 captured_inputs_(captured_inputs) {} in BorrowedArgsCallFrame()378 return args_.size() + captured_inputs_->size(); in num_args()384 const int captured_inputs_size = captured_inputs_->size(); in GetArg()389 *val = &(*captured_inputs_)[index - args_size]; in GetArg()398 const std::vector<Tensor>* const captured_inputs_; // Not owned. member in tensorflow::data::__anon5a9b84710111::BorrowedArgsCallFrame[all …]
164 return captured_inputs_; in captured_inputs()195 const std::vector<Tensor> captured_inputs_; variable