Home
last modified time | relevance | path

Searched refs:captured_func_ (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_map_dataset_op.cc98 captured_func_(std::move(captured_func)), in Dataset()
146 other_arguments_types.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
148 other_arguments.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
149 for (const Tensor& t : captured_func_->captured_inputs()) { in AsGraphDefInternal()
215 dataset_->captured_func_->captured_inputs(); in MapFunc()
244 return dataset_->captured_func_->Instantiate( in InitFunc()
281 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::__anone56384da0111::ParallelMapDatasetOp::Dataset
Dfilter_dataset_op.cc103 captured_func_(std::move(captured_func)), in Dataset()
135 other_arguments_types.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
137 other_arguments.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
138 for (const Tensor& t : captured_func_->captured_inputs()) { in AsGraphDefInternal()
176 return dataset()->captured_func_->Instantiate( in Initialize()
290 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::__anon68599af60111::FilterDatasetOp::Dataset
Dflat_map_dataset_op.cc60 captured_func_(std::move(captured_func)), in Dataset()
93 other_arguments_types.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
95 other_arguments.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
96 for (const Tensor& t : captured_func_->captured_inputs()) { in AsGraphDefInternal()
131 return dataset()->captured_func_->Instantiate( in Initialize()
266 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::__anondb4b4ff40111::FlatMapDatasetOp::Dataset
Dmap_dataset_op.cc110 captured_func_(std::move(captured_func)), in Dataset()
144 other_arguments_types.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
146 other_arguments.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
147 for (const Tensor& t : captured_func_->captured_inputs()) { in AsGraphDefInternal()
200 return dataset()->captured_func_->Instantiate( in Initialize()
267 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::__anon1ccfd74e0111::MapDatasetOp::Dataset
Dinterleave_dataset_op.cc80 captured_func_(std::move(captured_func)), in Dataset()
119 other_arguments_types.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
121 other_arguments.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
122 for (const Tensor& t : captured_func_->captured_inputs()) { in AsGraphDefInternal()
158 return dataset()->captured_func_->Instantiate( in Initialize()
321 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::__anon66e7b9ff0111::InterleaveDatasetOp::Dataset
Dcaptured_function.cc291 captured_func_(captured_func) {} in InstantiatedCapturedFunction()
323 OwnedArgsCallFrame frame(std::move(args), &captured_func_->captured_inputs(), in Run()
359 BorrowedArgsCallFrame frame(args, &captured_func_->captured_inputs(), in RunWithBorrowedArgs()
395 BorrowedArgsCallFrame frame(args, &captured_func_->captured_inputs(), in RunInstantiated()
416 std::move(args), &captured_func_->captured_inputs(), ret_types_); in RunAsync()
464 "::", captured_func_->func().name()); in RunAsync()
Dparallel_interleave_dataset_op.cc116 captured_func_(std::move(captured_func)), in Dataset()
163 other_arguments_types.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
165 other_arguments.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
166 for (const Tensor& t : captured_func_->captured_inputs()) { in AsGraphDefInternal()
235 return dataset()->captured_func_->Instantiate( in Initialize()
923 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::__anon1cf7320b0111::ParallelInterleaveDatasetOp::Dataset
Dcaptured_function.h107 CapturedFunction* const captured_func_; variable
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dtake_while_dataset_op.cc101 captured_func_(std::move(captured_func)), in Dataset()
138 other_arguments.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
140 other_arguments_types.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
141 for (const Tensor& t : captured_func_->captured_inputs()) { in AsGraphDefInternal()
177 return dataset()->captured_func_->Instantiate( in Initialize()
238 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::__anon850bf76a0111::TakeWhileDatasetOp::Dataset
Dscan_dataset_op.cc75 captured_func_(std::move(captured_func)), in Dataset()
117 other_arguments.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
119 other_arguments_types.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
120 for (const Tensor& t : captured_func_->captured_inputs()) { in AsGraphDefInternal()
161 return dataset()->captured_func_->Instantiate( in Initialize()
288 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::__anonff9a9bf20111::ScanDatasetOp::Dataset
Dmap_and_batch_dataset_op.cc162 captured_func_(std::move(captured_func)), in Dataset()
215 other_arguments_types.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
217 other_arguments.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
218 for (const Tensor& t : captured_func_->captured_inputs()) { in AsGraphDefInternal()
287 return dataset()->captured_func_->Instantiate( in Initialize()
817 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::__anonba85ab5e0111::MapAndBatchDatasetOp::Dataset
Dparallel_interleave_dataset_op.cc100 captured_func_(std::move(captured_func)), in Dataset()
151 other_arguments_types.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
153 other_arguments.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
154 for (const Tensor& t : captured_func_->captured_inputs()) { in AsGraphDefInternal()
256 return dataset()->captured_func_->Instantiate( in Initialize()
1069 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::__anon80c1a0a80111::ParallelInterleaveDatasetOp::Dataset
Dnuma_map_and_batch_dataset_op.cc115 captured_func_(std::move(captured_func)) { in Dataset()
166 other_arguments_types.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
168 other_arguments.reserve(captured_func_->captured_inputs().size()); in AsGraphDefInternal()
169 for (const Tensor& t : captured_func_->captured_inputs()) { in AsGraphDefInternal()
228 TF_RETURN_IF_ERROR(dataset()->captured_func_->Instantiate( in Initialize()
1145 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::__anon1c159b6e0111::NumaMapAndBatchDatasetOp::Dataset