Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.h37 class CapturedFunction; variable
99 CapturedFunction* captured_func);
101 friend class CapturedFunction; variable
107 CapturedFunction* const captured_func_;
114 class CapturedFunction {
120 std::unique_ptr<CapturedFunction>* out_function);
130 std::unique_ptr<CapturedFunction>* out_function);
140 std::unique_ptr<CapturedFunction>* out_function);
155 CapturedFunction(const NameAttrList& func,
163 TF_DISALLOW_COPY_AND_ASSIGN(CapturedFunction);
[all …]
Dgenerator_dataset_op.cc33 Dataset(OpKernelContext* ctx, std::unique_ptr<CapturedFunction> init_func, in Dataset()
34 std::unique_ptr<CapturedFunction> next_func, in Dataset()
35 std::unique_ptr<CapturedFunction> finalize_func, in Dataset()
148 const std::unique_ptr<CapturedFunction> init_func_;
149 const std::unique_ptr<CapturedFunction> next_func_;
150 const std::unique_ptr<CapturedFunction> finalize_func_;
166 std::unique_ptr<CapturedFunction> init_func; in MakeDataset()
167 OP_REQUIRES_OK(ctx, CapturedFunction::Create( in MakeDataset()
170 std::unique_ptr<CapturedFunction> next_func; in MakeDataset()
171 OP_REQUIRES_OK(ctx, CapturedFunction::Create( in MakeDataset()
[all …]
Dcaptured_function.cc104 Status CapturedFunction::Create( in Create()
106 std::unique_ptr<CapturedFunction>* out_function) { in Create()
107 return CapturedFunction::Create(func, ctx, argument_name, true, out_function); in Create()
110 Status CapturedFunction::Create( in Create()
113 std::unique_ptr<CapturedFunction>* out_function) { in Create()
117 *out_function = absl::WrapUnique(new CapturedFunction( in Create()
122 Status CapturedFunction::Create( in Create()
125 std::unique_ptr<CapturedFunction>* out_function) { in Create()
126 *out_function = absl::WrapUnique(new CapturedFunction( in Create()
131 Status CapturedFunction::Instantiate( in Instantiate()
[all …]
Dmap_dataset_op.cc49 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
50 OP_REQUIRES_OK(ctx, CapturedFunction::Create(func_, ctx, "other_arguments", in MakeDataset()
55 CapturedFunction* raw_captured_func = captured_func.get(); in MakeDataset()
100 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
267 const std::unique_ptr<CapturedFunction> captured_func_;
Dflat_map_dataset_op.cc42 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
43 OP_REQUIRES_OK(ctx, CapturedFunction::Create(func_, ctx, "other_arguments", in MakeDataset()
54 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
266 const std::unique_ptr<CapturedFunction> captured_func_;
Dfilter_dataset_op.cc52 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
53 OP_REQUIRES_OK(ctx, CapturedFunction::Create(func_, ctx, "other_arguments", in MakeDataset()
98 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
290 const std::unique_ptr<CapturedFunction> captured_func_;
Dparallel_map_dataset_op.cc64 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
65 OP_REQUIRES_OK(ctx, CapturedFunction::Create(func_, ctx, "other_arguments", in MakeDataset()
87 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
281 const std::unique_ptr<CapturedFunction> captured_func_;
Dinterleave_dataset_op.cc60 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
61 OP_REQUIRES_OK(ctx, CapturedFunction::Create(func_, ctx, "other_arguments", in MakeDataset()
74 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length, in Dataset()
321 const std::unique_ptr<CapturedFunction> captured_func_;
Dparallel_interleave_dataset_op.cc89 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
91 ctx, CapturedFunction::Create(interleave_func_, ctx, "other_arguments", in MakeDataset()
109 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length, in Dataset()
923 const std::unique_ptr<CapturedFunction> captured_func_;
Diterator_ops.cc663 std::unique_ptr<CapturedFunction> captured_func; in ComputeAsync()
666 CapturedFunction::Create(reduce_func_, ctx, "other_arguments", in ComputeAsync()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dgroup_by_reducer_dataset_op.cc44 std::unique_ptr<CapturedFunction> captured_key_func; in MakeDataset()
45 OP_REQUIRES_OK(ctx, CapturedFunction::Create(key_func_, ctx, in MakeDataset()
48 std::unique_ptr<CapturedFunction> captured_init_func; in MakeDataset()
49 OP_REQUIRES_OK(ctx, CapturedFunction::Create(init_func_, ctx, in MakeDataset()
52 std::unique_ptr<CapturedFunction> captured_reduce_func; in MakeDataset()
53 OP_REQUIRES_OK(ctx, CapturedFunction::Create(reduce_func_, ctx, in MakeDataset()
56 std::unique_ptr<CapturedFunction> captured_finalize_func; in MakeDataset()
58 CapturedFunction::Create(finalize_func_, ctx, in MakeDataset()
72 std::unique_ptr<CapturedFunction> captured_key_func, in Dataset()
73 std::unique_ptr<CapturedFunction> captured_init_func, in Dataset()
[all …]
Dgroup_by_window_dataset_op.cc44 std::unique_ptr<CapturedFunction> captured_key_func; in MakeDataset()
45 OP_REQUIRES_OK(ctx, CapturedFunction::Create(key_func_, ctx, in MakeDataset()
48 std::unique_ptr<CapturedFunction> captured_reduce_func; in MakeDataset()
49 OP_REQUIRES_OK(ctx, CapturedFunction::Create(reduce_func_, ctx, in MakeDataset()
52 std::unique_ptr<CapturedFunction> captured_window_size_func; in MakeDataset()
54 CapturedFunction::Create(window_size_func_, ctx, in MakeDataset()
70 std::unique_ptr<CapturedFunction> captured_key_func, in Dataset()
71 std::unique_ptr<CapturedFunction> captured_reduce_func, in Dataset()
72 std::unique_ptr<CapturedFunction> captured_window_size_func, in Dataset()
495 const std::unique_ptr<CapturedFunction>& captured_func, in OtherArgumentsNodeAndType()
[all …]
Dtake_while_dataset_op.cc51 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
52 OP_REQUIRES_OK(ctx, CapturedFunction::Create(func_, ctx, "other_arguments", in MakeDataset()
96 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
238 const std::unique_ptr<CapturedFunction> captured_func_;
Dscan_dataset_op.cc52 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
53 OP_REQUIRES_OK(ctx, CapturedFunction::Create(func_, ctx, "other_arguments", in MakeDataset()
66 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
288 const std::unique_ptr<CapturedFunction> captured_func_;
Dchoose_fastest_branch_dataset_op.cc161 std::vector<std::unique_ptr<CapturedFunction>> captured_funcs( in MakeDataset()
176 ctx, CapturedFunction::Create( in MakeDataset()
191 std::vector<std::unique_ptr<CapturedFunction>> captured_funcs, in Dataset()
534 const std::vector<std::unique_ptr<CapturedFunction>> captured_funcs_;
Dmap_and_batch_dataset_op.cc88 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
89 OP_REQUIRES_OK(ctx, CapturedFunction::Create(func_, ctx, "other_arguments", in MakeDataset()
93 CapturedFunction* raw_captured_func = captured_func.get(); in MakeDataset()
151 std::unique_ptr<CapturedFunction> captured_func, in Dataset()
817 const std::unique_ptr<CapturedFunction> captured_func_;
Dparallel_interleave_dataset_op.cc77 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
79 ctx, CapturedFunction::Create(interleave_func_, ctx, "other_arguments", in MakeDataset()
93 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length, in Dataset()
1069 const std::unique_ptr<CapturedFunction> captured_func_;
Dnuma_map_and_batch_dataset_op.cc87 std::unique_ptr<CapturedFunction> captured_func; in MakeDataset()
89 ctx, CapturedFunction::Create(func_, ctx, "other_arguments", in MakeDataset()
106 std::unique_ptr<CapturedFunction> captured_func) in Dataset()
1145 const std::unique_ptr<CapturedFunction> captured_func_;