Home
last modified time | relevance | path

Searched refs:captured_inputs (Results 1 – 25 of 50) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/
Dmap_defun_op.cc139 OpInputList captured_inputs; member
151 OpInputList captured_inputs, in ComputeOptions()
157 captured_inputs(captured_inputs), in ComputeOptions()
182 compute_opts_->captured_inputs.size()) { in GetArg()
190 compute_opts_->captured_inputs[index - compute_opts_->args.size()]; in GetArg()
269 OpInputList captured_inputs; in SetupArgs() local
270 TF_RETURN_IF_ERROR(ctx->input_list("captured_inputs", &captured_inputs)); in SetupArgs()
296 ctx, arguments, captured_inputs, std::move(arg_shapes), batch_size, in SetupArgs()
Dcaptured_function.cc124 std::vector<Tensor>&& captured_inputs, bool use_inter_op_parallelism, in Create() argument
127 func, std::move(captured_inputs), use_inter_op_parallelism)); in Create()
214 const std::vector<Tensor>* captured_inputs, in OwnedArgsCallFrame() argument
218 captured_inputs_(captured_inputs) {} in OwnedArgsCallFrame()
250 const std::vector<Tensor>* captured_inputs, in BorrowedArgsCallFrame() argument
254 captured_inputs_(captured_inputs) {} in BorrowedArgsCallFrame()
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()
[all …]
Dparallel_map_dataset_op.cc146 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()
214 const std::vector<Tensor>& captured_inputs = in MapFunc() local
215 dataset_->captured_func_->captured_inputs(); in MapFunc()
228 captured_inputs[dataset_->short_circuit_indices_[i] - in MapFunc()
Dmap_dataset_op.cc71 const std::vector<Tensor>& captured_inputs = in MakeDataset() local
72 raw_captured_func->captured_inputs(); in MakeDataset()
82 out_tensors->push_back(captured_inputs[indices[i] - num_args]); in MakeDataset()
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()
Dcaptured_function.h138 std::vector<Tensor>&& captured_inputs,
152 const std::vector<Tensor>& captured_inputs() { return captured_inputs_; } in captured_inputs() function
156 std::vector<Tensor> captured_inputs,
Dflat_map_dataset_op.cc93 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()
Dfilter_dataset_op.cc135 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()
Dinterleave_dataset_op.cc119 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()
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dprefetching_ops.py126 args=init_func_concrete.captured_inputs,
131 self._init_captured_args = self._init_func.captured_inputs
159 next_func_concrete.captured_inputs,
164 self._next_captured_args = self._next_func.captured_inputs
189 args=[string_handle] + finalize_func_concrete.captured_inputs,
195 self._finalize_captured_args = self._finalize_func.captured_inputs
243 self._map_func.function.captured_inputs,
Dgrouping.py258 self._key_func.function.captured_inputs,
259 self._init_func.function.captured_inputs,
260 self._reduce_func.function.captured_inputs,
261 self._finalize_func.function.captured_inputs,
381 self._key_func.function.captured_inputs,
382 self._reduce_func.function.captured_inputs,
383 self._window_size_func.function.captured_inputs,
Doptimization.py265 self._captured_arguments.extend(f.function.captured_inputs)
267 len(f.function.captured_inputs) for f in self._funcs
Dmap_defun.py66 return gen_dataset_ops.map_defun(elems, concrete_fn.captured_inputs,
Dtake_while_ops.py46 other_arguments=self._predicate.function.captured_inputs,
/external/tensorflow/tensorflow/python/data/ops/
Dmulti_device_iterator_ops.py57 args=init_func_concrete.captured_inputs,
62 self._init_captured_args = self._init_func.captured_inputs
92 args=[string_handle] + next_func_concrete.captured_inputs,
97 self._next_captured_args = self._next_func.captured_inputs
120 args=[string_handle] + finalize_func_concrete.captured_inputs,
126 self._finalize_captured_args = self._finalize_func.captured_inputs
159 self._init_captured_args = self._init_func.captured_inputs
Ddataset_ops.py142 if any([is_tensor_or_parent_ref(t) for t in fn.function.captured_inputs]):
1348 reduce_func.captured_inputs,
2494 + self._init_func.function.captured_inputs,
2495 self._next_func.function.captured_inputs,
2496 self._finalize_func.function.captured_inputs,
3008 self._map_func.function.captured_inputs,
3049 self._map_func.function.captured_inputs,
3083 self._map_func.function.captured_inputs,
3120 self._map_func.function.captured_inputs, # pylint: disable=protected-access
3160 self._map_func.function.captured_inputs, # pylint: disable=protected-access
[all …]
/external/tensorflow/tensorflow/python/ops/
Dfunctional_ops.py610 if cond.captured_inputs:
619 if body.captured_inputs:
621 body.declared_input_types) + [t.dtype for t in body.captured_inputs]
629 input_ + body.captured_inputs,
634 ret = ret[:-len(body.captured_inputs)]
753 if body.captured_inputs:
758 inputs + body.captured_inputs,
762 ret = ret[:-len(body.captured_inputs)]
/external/tensorflow/tensorflow/python/framework/
Dconvert_to_constants.py108 output_func._captured_inputs = input_func.captured_inputs # pylint: disable=protected-access
149 input_tensors = func.inputs[-len(func.captured_inputs):]
151 index = func.captured_inputs.index(var.handle)
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dtake_while_dataset_op.cc138 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()
Dscan_dataset_op.cc117 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()
Dmap_and_batch_dataset_op.cc112 const std::vector<Tensor>& captured_inputs = in MakeDataset() local
113 raw_captured_func->captured_inputs(); in MakeDataset()
123 out_tensors->push_back(captured_inputs[indices[i] - num_args]); in MakeDataset()
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()
Dgroup_by_reducer_dataset_op.cc413 other_arguments_node->reserve(captured_func->captured_inputs().size()); in OtherArgumentsNodeAndType()
414 other_arguments_types->reserve(captured_func->captured_inputs().size()); in OtherArgumentsNodeAndType()
415 for (const Tensor& t : captured_func->captured_inputs()) { in OtherArgumentsNodeAndType()
Dchoose_fastest_branch_dataset_op.cc256 num_captured_inputs += func->captured_inputs().size(); in AsGraphDefInternal()
257 other_arguments_lengths.push_back(func->captured_inputs().size()); in AsGraphDefInternal()
264 for (const Tensor& t : func->captured_inputs()) { in AsGraphDefInternal()
/external/tensorflow/tensorflow/contrib/batching/python/ops/
Dbatch_ops_test.py279 captured_tensors=computation.captured_inputs)
311 captured_tensors=computation.captured_inputs,
342 captured_tensors=computation.captured_inputs,
/external/tensorflow/tensorflow/python/saved_model/
Dfunction_serialization.py44 for capture in concrete_function.captured_inputs:
/external/tensorflow/tensorflow/core/kernels/
Dbatch_kernels.cc266 batch_components->captured_inputs.push_back(captured_tensor); in RegisterInput()
287 std::vector<Tensor> captured_inputs; member
523 const auto& captured_inputs = in ProcessFuncBatch() local
524 batch->task(batch->num_tasks() - 1).captured_inputs; in ProcessFuncBatch()
525 args.insert(args.end(), captured_inputs.begin(), captured_inputs.end()); in ProcessFuncBatch()

12