Home
last modified time | relevance | path

Searched refs:CapturedFunction (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dgenerator_dataset_op.cc49 Dataset(OpKernelContext* ctx, std::unique_ptr<CapturedFunction> init_func, in Dataset()
50 std::unique_ptr<CapturedFunction> next_func, in Dataset()
51 std::unique_ptr<CapturedFunction> finalize_func, in Dataset()
187 const std::unique_ptr<CapturedFunction> init_func_;
188 const std::unique_ptr<CapturedFunction> next_func_;
189 const std::unique_ptr<CapturedFunction> finalize_func_;
209 std::unique_ptr<CapturedFunction> init_func; in MakeDataset()
210 OP_REQUIRES_OK(ctx, CapturedFunction::Create(ctx, init_func_metadata_, in MakeDataset()
213 std::unique_ptr<CapturedFunction> next_func; in MakeDataset()
214 OP_REQUIRES_OK(ctx, CapturedFunction::Create(ctx, next_func_metadata_, in MakeDataset()
[all …]
Dcaptured_function.h40 class CapturedFunction; variable
129 class CapturedFunction {
136 std::unique_ptr<CapturedFunction>* out_function);
143 std::unique_ptr<CapturedFunction>* out_function);
189 CapturedFunction(std::shared_ptr<const FunctionMetadata> metadata,
197 TF_DISALLOW_COPY_AND_ASSIGN(CapturedFunction);
218 CapturedFunction* captured_func, bool is_multi_device,
277 CapturedFunction* captured_func, bool is_multi_device);
289 CapturedFunction* const captured_func_; // Not owned.
Dcaptured_function.cc113 Status GetCapturedInput(const CapturedFunction* const func, int index, in GetCapturedInput()
127 const CapturedFunction* const func, in RunShortCircuit()
146 const CapturedFunction* const func, in RunShortCircuit()
544 Status CapturedFunction::Create( in Create()
547 std::unique_ptr<CapturedFunction>* out_function) { in Create()
556 Status CapturedFunction::Create( in Create()
559 std::unique_ptr<CapturedFunction>* out_function) { in Create()
561 new CapturedFunction(std::move(metadata), std::move(captured_inputs))); in Create()
565 Status CapturedFunction::AddToGraph( in AddToGraph()
588 Status CapturedFunction::Instantiate( in Instantiate()
[all …]
Dmap_dataset_op.cc44 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
209 const std::unique_ptr<CapturedFunction> captured_func_;
229 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
231 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
Dfilter_dataset_op.cc51 std::unique_ptr<CapturedFunction> captured_func) in Dataset()
242 const std::unique_ptr<CapturedFunction> captured_func_;
256 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
258 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
Dflat_map_dataset_op.cc49 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
270 const std::unique_ptr<CapturedFunction> captured_func_;
285 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
287 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
Dinterleave_dataset_op.cc54 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length, in Dataset()
304 const std::unique_ptr<CapturedFunction> captured_func_;
337 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
339 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
Dparallel_map_dataset_op.cc78 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
671 const std::unique_ptr<CapturedFunction> captured_func_;
721 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
723 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
Dparallel_interleave_dataset_op.cc162 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length, in Dataset()
1557 const std::unique_ptr<CapturedFunction> captured_func_;
1655 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
1657 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
Diterator_ops.cc702 std::unique_ptr<CapturedFunction> captured_func; in DoCompute()
703 TF_RETURN_IF_ERROR(CapturedFunction::Create( in DoCompute()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dgroup_by_reducer_dataset_op.cc52 std::unique_ptr<CapturedFunction> captured_key_func; in MakeDataset()
53 OP_REQUIRES_OK(ctx, CapturedFunction::Create(ctx, key_func_metadata_, in MakeDataset()
56 std::unique_ptr<CapturedFunction> captured_init_func; in MakeDataset()
57 OP_REQUIRES_OK(ctx, CapturedFunction::Create(ctx, init_func_metadata_, in MakeDataset()
60 std::unique_ptr<CapturedFunction> captured_reduce_func; in MakeDataset()
61 OP_REQUIRES_OK(ctx, CapturedFunction::Create(ctx, reduce_func_metadata_, in MakeDataset()
64 std::unique_ptr<CapturedFunction> captured_finalize_func; in MakeDataset()
66 CapturedFunction::Create(ctx, finalize_func_metadata_, in MakeDataset()
80 std::unique_ptr<CapturedFunction> captured_key_func, in Dataset()
81 std::unique_ptr<CapturedFunction> captured_init_func, in Dataset()
[all …]
Dgroup_by_window_dataset_op.cc50 std::unique_ptr<CapturedFunction> captured_key_func; in MakeDataset()
51 OP_REQUIRES_OK(ctx, CapturedFunction::Create(ctx, key_func_metadata_, in MakeDataset()
55 std::unique_ptr<CapturedFunction> captured_reduce_func; in MakeDataset()
56 OP_REQUIRES_OK(ctx, CapturedFunction::Create(ctx, reduce_func_metadata_, in MakeDataset()
60 std::unique_ptr<CapturedFunction> captured_window_size_func; in MakeDataset()
62 CapturedFunction::Create(ctx, window_size_func_metadata_, in MakeDataset()
76 std::unique_ptr<CapturedFunction> captured_key_func, in Dataset()
77 std::unique_ptr<CapturedFunction> captured_reduce_func, in Dataset()
78 std::unique_ptr<CapturedFunction> captured_window_size_func, in Dataset()
530 const std::unique_ptr<CapturedFunction> captured_key_func_;
[all …]
Dio_ops.cc59 std::unique_ptr<CapturedFunction> captured_func; in DoCompute()
60 TF_RETURN_IF_ERROR(CapturedFunction::Create( in DoCompute()
73 std::unique_ptr<CapturedFunction> captured_func, in WriteData()
182 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
339 const std::unique_ptr<CapturedFunction> captured_func_;
359 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
361 ctx, CapturedFunction::Create(ctx, func_metadata_, kReaderFuncOtherArgs, in MakeDataset()
Dtake_while_dataset_op.cc45 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
47 ctx, CapturedFunction::Create(ctx, func_metadata_, "other_arguments", in MakeDataset()
56 std::unique_ptr<CapturedFunction> captured_func) in Dataset()
205 const std::unique_ptr<CapturedFunction> captured_func_;
Dscan_dataset_op.cc59 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
61 ctx, CapturedFunction::Create(ctx, func_metadata_, "other_arguments", in MakeDataset()
75 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
306 const std::unique_ptr<CapturedFunction> captured_func_;
Dchoose_fastest_branch_dataset_op.cc178 std::vector<std::unique_ptr<CapturedFunction>> captured_funcs( in MakeDataset()
192 OP_REQUIRES_OK(ctx, CapturedFunction::Create(ctx, func_metadatas_[i], in MakeDataset()
205 std::vector<std::unique_ptr<CapturedFunction>> captured_funcs, in Dataset()
568 const std::vector<std::unique_ptr<CapturedFunction>> captured_funcs_;
Dio_ops.h53 std::unique_ptr<CapturedFunction> captured_func,
Dmap_and_batch_dataset_op.cc88 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
686 const std::unique_ptr<CapturedFunction> captured_func_;
719 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
721 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
Dsnapshot_dataset_op.cc123 std::unique_ptr<CapturedFunction> reader_func,
124 std::unique_ptr<CapturedFunction> shard_func);
156 std::unique_ptr<CapturedFunction> reader_func_;
157 std::unique_ptr<CapturedFunction> shard_func_;
312 std::unique_ptr<CapturedFunction> reader_func, in Dataset()
313 std::unique_ptr<CapturedFunction> shard_func) in Dataset()
887 std::unique_ptr<CapturedFunction> reader_func; in MakeDataset()
889 CapturedFunction::Create(ctx, reader_func_metadata_, in MakeDataset()
891 std::unique_ptr<CapturedFunction> shard_func; in MakeDataset()
893 CapturedFunction::Create(ctx, shard_func_metadata_, in MakeDataset()
Dparallel_interleave_dataset_op.cc96 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length, in Dataset()
1160 const std::unique_ptr<CapturedFunction> captured_func_;
1226 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
1228 CapturedFunction::Create(ctx, func_metadata_, kOtherArguments, in MakeDataset()
/external/tensorflow/tensorflow/core/framework/
Ddataset.h910 friend class CapturedFunction; variable