Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Dmap_dataset_op.cc51 captured_func_(std::move(captured_func)), in Dataset()
89 TF_RETURN_IF_ERROR(captured_func_->CheckExternalState()); in CheckExternalState()
102 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
107 b->BuildAttrValue(captured_func_->func(), &f_attr); in AsGraphDefInternal()
115 b->BuildAttrValue(captured_func_->use_inter_op_parallelism(), in AsGraphDefInternal()
143 return dataset()->captured_func_->Instantiate( in Initialize()
191 dataset()->captured_func_->CheckExternalState())); in SaveInternal()
209 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::MapDatasetOp::Dataset
Dfilter_dataset_op.cc54 captured_func_(std::move(captured_func)) { in Dataset()
84 TF_RETURN_IF_ERROR(captured_func_->CheckExternalState()); in CheckExternalState()
96 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
99 b->BuildAttrValue(captured_func_->func(), &f); in AsGraphDefInternal()
120 return dataset()->captured_func_->Instantiate( in Initialize()
205 dataset()->captured_func_->CheckExternalState())); in SaveInternal()
242 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::FilterDatasetOp::Dataset
Dflat_map_dataset_op.cc54 captured_func_(std::move(captured_func)), in Dataset()
84 TF_RETURN_IF_ERROR(captured_func_->CheckExternalState()); in CheckExternalState()
96 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
99 b->BuildAttrValue(captured_func_->func(), &f); in AsGraphDefInternal()
121 return dataset()->captured_func_->Instantiate( in Initialize()
174 dataset()->captured_func_->CheckExternalState())); in SaveInternal()
270 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::FlatMapDatasetOp::Dataset
Dinterleave_dataset_op.cc59 captured_func_(std::move(captured_func)), in Dataset()
96 TF_RETURN_IF_ERROR(captured_func_->CheckExternalState()); in CheckExternalState()
112 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
115 b->BuildAttrValue(captured_func_->func(), &f); in AsGraphDefInternal()
137 return dataset()->captured_func_->Instantiate( in Initialize()
206 dataset()->captured_func_->CheckExternalState())); in SaveInternal()
304 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::InterleaveDatasetOp::Dataset
Dcaptured_function.cc824 captured_func_(captured_func), in InstantiatedCapturedFunction()
836 auto& info = captured_func_->short_circuit_info(); in Run()
838 return RunShortCircuit(info, std::move(args), captured_func_, rets); in Run()
860 OwnedArgsCallFrame frame(std::move(args), &captured_func_->captured_inputs(), in Run()
876 node->name(), stats_utils::kDelimiter, captured_func_->func().name()); in Run()
899 auto& info = captured_func_->short_circuit_info(); in RunWithBorrowedArgs()
901 return RunShortCircuit(info, args, captured_func_, rets); in RunWithBorrowedArgs()
923 BorrowedArgsCallFrame frame(args, &captured_func_->captured_inputs(), in RunWithBorrowedArgs()
938 node->name(), stats_utils::kDelimiter, captured_func_->func().name()); in RunWithBorrowedArgs()
954 auto& info = captured_func_->short_circuit_info(); in RunInstantiated()
[all …]
Dparallel_map_dataset_op.cc87 captured_func_(std::move(captured_func)), in Dataset()
129 TF_RETURN_IF_ERROR(captured_func_->CheckExternalState()); in CheckExternalState()
144 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
160 b->BuildAttrValue(captured_func_->func(), &f_attr); in AsGraphDefInternal()
170 b->BuildAttrValue(captured_func_->use_inter_op_parallelism(), in AsGraphDefInternal()
235 return dataset()->captured_func_->Instantiate( in Initialize()
278 dataset()->captured_func_->CheckExternalState())); in SaveInternal()
442 if (dataset()->captured_func_->use_inter_op_parallelism()) { in CallFunction()
671 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::ParallelMapDatasetOp::Dataset
Dparallel_interleave_dataset_op.cc169 captured_func_(std::move(captured_func)), in Dataset()
236 TF_RETURN_IF_ERROR(captured_func_->CheckExternalState()); in CheckExternalState()
254 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
285 b->BuildAttrValue(captured_func_->func(), &f); in AsGraphDefInternal()
352 return dataset()->captured_func_->Instantiate( in Initialize()
409 dataset()->captured_func_->CheckExternalState())); in SaveInternal()
1557 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::ParallelInterleaveDatasetOp::Dataset
Dcaptured_function.h289 CapturedFunction* const captured_func_; // Not owned. variable
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dtake_while_dataset_op.cc59 captured_func_(std::move(captured_func)) { in Dataset()
92 TF_RETURN_IF_ERROR(captured_func_->CheckExternalState()); in CheckExternalState()
105 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
108 b->BuildAttrValue(captured_func_->func(), &f_attr); in AsGraphDefInternal()
131 return dataset()->captured_func_->Instantiate( in Initialize()
178 dataset()->captured_func_->CheckExternalState())); in SaveInternal()
205 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::experimental::__anonaa7f23910111::TakeWhileDatasetOp::Dataset
Dscan_dataset_op.cc83 captured_func_(std::move(captured_func)), in Dataset()
124 TF_RETURN_IF_ERROR(captured_func_->CheckExternalState()); in CheckExternalState()
143 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
146 b->BuildAttrValue(captured_func_->func(), &f); in AsGraphDefInternal()
177 return dataset()->captured_func_->Instantiate( in Initialize()
266 dataset()->captured_func_->CheckExternalState())); in SaveInternal()
306 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::experimental::__anon700576d90111::ScanDatasetOp::Dataset
Dio_ops.cc186 captured_func_(std::move(captured_func)), in Dataset()
212 return captured_func_->CheckExternalState(); in CheckExternalState()
224 TF_RETURN_IF_ERROR(captured_func_->AddToGraph( in AsGraphDefInternal()
233 b->BuildAttrValue(captured_func_->func(), &reader_func_attr); in AsGraphDefInternal()
260 TF_RETURN_IF_ERROR(dataset()->captured_func_->Instantiate( in Initialize()
339 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::experimental::LoadDatasetOp::Dataset
Dmap_and_batch_dataset_op.cc97 captured_func_(std::move(captured_func)), in Dataset()
143 TF_RETURN_IF_ERROR(captured_func_->CheckExternalState()); in CheckExternalState()
162 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
165 b->BuildAttrValue(captured_func_->func(), &f); in AsGraphDefInternal()
226 return dataset()->captured_func_->Instantiate( in Initialize()
281 dataset()->captured_func_->CheckExternalState())); in SaveInternal()
686 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::experimental::MapAndBatchDatasetOp::Dataset
Dparallel_interleave_dataset_op.cc103 captured_func_(std::move(captured_func)), in Dataset()
156 TF_RETURN_IF_ERROR(captured_func_->CheckExternalState()); in CheckExternalState()
174 TF_RETURN_IF_ERROR(captured_func_->AddToGraph(ctx, b, &other_arguments, in AsGraphDefInternal()
206 b->BuildAttrValue(captured_func_->func(), &f); in AsGraphDefInternal()
289 return dataset()->captured_func_->Instantiate( in Initialize()
414 dataset()->captured_func_->CheckExternalState())); in SaveInternal()
1160 const std::unique_ptr<CapturedFunction> captured_func_; member in tensorflow::data::experimental::ParallelInterleaveDatasetOp::Dataset