| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 49 OpInputList captured_inputs; member
61 OpInputList captured_inputs, in ComputeOptions()
67 captured_inputs(captured_inputs), in ComputeOptions()
95 compute_opts_->captured_inputs.size()) { in GetArg()
102 &compute_opts_->captured_inputs[index - compute_opts_->args.size()]; in GetArg()
259 OpInputList captured_inputs; in SetupArgs() local
260 TF_RETURN_IF_ERROR(ctx->input_list(kCapturedInputs, &captured_inputs)); in SetupArgs()
286 new ComputeOptions(ctx, arguments, captured_inputs, std::move(arg_shapes), in SetupArgs()
|
| D | captured_function.cc | 115 if (TF_PREDICT_FALSE(index >= func->captured_inputs().size())) { in GetCapturedInput()
119 ". Num captured inputs: ", func->captured_inputs().size()); in GetCapturedInput()
121 *out = &func->captured_inputs()[index]; in GetCapturedInput()
327 const std::vector<Tensor>* captured_inputs, in OwnedArgsCallFrame() argument
331 captured_inputs_(captured_inputs) {} in OwnedArgsCallFrame()
371 const std::vector<Tensor>* captured_inputs, in BorrowedArgsCallFrame() argument
375 captured_inputs_(captured_inputs) {} in BorrowedArgsCallFrame()
550 std::vector<Tensor> captured_inputs(inputs.begin(), inputs.end()); in Create() local
551 return Create(ctx, std::move(metadata), std::move(captured_inputs), in Create()
558 std::vector<Tensor>&& captured_inputs, in Create() argument
[all …]
|
| D | captured_function.h | 142 std::vector<Tensor>&& captured_inputs,
163 const std::vector<Tensor>& captured_inputs() const { in captured_inputs() function
190 std::vector<Tensor> captured_inputs);
|
| D | map_defun_op_test.cc | 26 std::vector<Tensor> captured_inputs, in MapDefunOpParams() argument
36 captured_inputs_(std::move(captured_inputs)), in MapDefunOpParams()
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | prefetching_ops.py | 127 args=init_func_concrete.captured_inputs,
132 self._init_captured_args = self._init_func.captured_inputs
159 args=[string_handle] + 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,
|
| D | grouping.py | 284 self._key_func.function.captured_inputs,
285 self._init_func.function.captured_inputs,
286 self._reduce_func.function.captured_inputs,
287 self._finalize_func.function.captured_inputs,
406 self._key_func.function.captured_inputs,
407 self._reduce_func.function.captured_inputs,
408 self._window_size_func.function.captured_inputs,
|
| D | optimization.py | 196 self._captured_arguments.extend(f.function.captured_inputs)
198 len(f.function.captured_inputs) for f in self._funcs
|
| D | io.py | 115 shard_func_other_args=shard_func.captured_inputs,
154 reader_func_other_args=self._reader_func.function.captured_inputs,
|
| D | scan_ops.py | 131 self._scan_func.function.captured_inputs,
140 self._scan_func.function.captured_inputs,
|
| D | map_defun.py | 66 return gen_dataset_ops.map_defun(elems, concrete_fn.captured_inputs,
|
| D | snapshot.py | 242 self._reader_func.function.captured_inputs,
243 self._shard_func.function.captured_inputs,
|
| D | take_while_ops.py | 46 other_arguments=self._predicate.function.captured_inputs,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | functional_ops.py | 897 num_non_captured_inputs = len(func.inputs) - len(func.captured_inputs)
951 if cond.captured_inputs:
963 if body.captured_inputs:
965 t.dtype for t in body.captured_inputs
974 input_ + body.captured_inputs,
979 ret = ret[:-len(body.captured_inputs)]
1098 if body.captured_inputs:
1103 inputs + body.captured_inputs,
1107 ret = ret[:-len(body.captured_inputs)]
|
| D | batch_ops.py | 117 captured_tensors=computation.captured_inputs,
|
| D | batch_ops_test.py | 366 captured_tensors=computation.captured_inputs)
400 captured_tensors=computation.captured_inputs,
433 captured_tensors=computation.captured_inputs,
464 captured_tensors=computation.captured_inputs)
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | multi_device_iterator_ops.py | 61 args=init_func_concrete.captured_inputs,
66 self._init_captured_args = self._init_func.captured_inputs
97 args=[string_handle] + next_func_concrete.captured_inputs,
102 self._next_captured_args = self._next_func.captured_inputs
125 args=[string_handle] + finalize_func_concrete.captured_inputs,
130 self._finalize_captured_args = self._finalize_func.captured_inputs
162 self._init_captured_args = self._init_func.captured_inputs
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MapDefun.pbtxt | 41 name: "captured_inputs"
87 name: "captured_inputs"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MapDefun.pbtxt | 41 name: "captured_inputs"
87 name: "captured_inputs"
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 138 task->captured_inputs = this->captured_inputs; in CreateSplitTask()
209 batch_components->captured_inputs.reserve(captured_tensors.size()); in RegisterInput()
211 batch_components->captured_inputs.push_back(captured_tensor); in RegisterInput()
601 const auto& captured_inputs = in ProcessFuncBatch() local
602 batch->task(batch->num_tasks() - 1).captured_inputs; in ProcessFuncBatch()
603 args.insert(args.end(), captured_inputs.begin(), captured_inputs.end()); in ProcessFuncBatch()
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | function_serialization.py | 64 for capture in concrete_function.captured_inputs:
161 …return inner_concrete._call_flat(args, inner_concrete.captured_inputs) # pylint:disable=protected…
|
| D | load.py | 93 def _call_flat(self, args, captured_inputs, cancellation_manager=None): argument
112 captured_inputs = list(map(get_handle, captured_inputs))
114 captured_inputs = list(map(get_unused_handle, captured_inputs))
115 return super(_WrapperFunction, self)._call_flat(args, captured_inputs,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MapDefun.pbtxt | 12 name: "captured_inputs"
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | def_function.py | 890 …filtered_flat_args, self._concrete_stateful_fn.captured_inputs) # pylint: disable=protected-access
943 captured_inputs=self._concrete_stateful_fn.captured_inputs))
1037 args=list(filtered_flat_args) + concrete_fn.captured_inputs)
|
| D | function.py | 763 for capture in backwards_function.captured_inputs
1215 captured_inputs = backward.captured_inputs
1218 for capture in captured_inputs
1232 backward_function_inputs = (len(backward.inputs) - len(captured_inputs))
1760 return self._call_flat(args, self.captured_inputs, cancellation_manager)
1785 captured_inputs=self.captured_inputs,
1868 def _call_flat(self, args, captured_inputs, cancellation_manager=None): argument
1937 args = tensor_inputs + captured_inputs
2027 def captured_inputs(self): member in ConcreteFunction
3001 …filtered_flat_args, captured_inputs=graph_function.captured_inputs) # pylint: disable=protected-a…
|
| /external/tensorflow/tensorflow/python/data/benchmarks/ |
| D | from_tensor_slices_benchmark.py | 44 self._map_func.function.captured_inputs,
|