Searched defs:captured_inputs (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 49 OpInputList captured_inputs; member
261 OpInputList captured_inputs; in SetupArgs() local
|
| D | map_defun_op_test.cc | 26 std::vector<Tensor> captured_inputs, in MapDefunOpParams()
|
| /external/tensorflow/tensorflow/distribute/experimental/rpc/kernels/ |
| D | rpc_ops.cc | 162 std::vector<Tensor> captured_inputs; member
183 std::vector<Tensor> captured_inputs, in Register()
241 std::vector<Tensor> captured_inputs = in Call() local
323 std::vector<Tensor> captured_inputs, in Register()
651 OpInputList captured_inputs; in Compute() local
|
| /external/tensorflow/tensorflow/core/data/ |
| D | captured_function.cc | 304 const std::vector<Tensor>* captured_inputs, in OwnedArgsCallFrame()
348 const std::vector<Tensor>* captured_inputs, in BorrowedArgsCallFrame()
488 std::vector<Tensor> captured_inputs(inputs.begin(), inputs.end()); in Create() local
496 std::vector<Tensor>&& captured_inputs, in Create()
692 std::vector<Tensor> captured_inputs) in CapturedFunction()
|
| D | captured_function.h | 188 const std::vector<Tensor>& captured_inputs() const { in captured_inputs() function
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.h | 72 std::vector<Tensor> captured_inputs; member
|
| D | batch_resource_base.cc | 721 const auto& captured_inputs = in ProcessFuncBatch() local
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load.py | 116 def _call_flat(self, args, captured_inputs, cancellation_manager=None): argument
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function.py | 388 def captured_inputs(self): member in _DefinedFunction
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function.py | 1756 def _call_flat(self, args, captured_inputs, cancellation_manager=None): argument
2032 def captured_inputs(self): member in ConcreteFunction
|