Home
last modified time | relevance | path

Searched defs:captured_inputs (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/distribute/experimental/rpc/kernels/
H A Drpc_ops.cc162 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
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dcaptured_function.cc304 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()
H A Dcaptured_function.h188 const std::vector<Tensor>& captured_inputs() const { in captured_inputs() function
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dmap_defun_op.cc49 OpInputList captured_inputs; member
261 OpInputList captured_inputs; in SetupArgs() local
H A Dmap_defun_op_test.cc26 std::vector<Tensor> captured_inputs, in MapDefunOpParams()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dbatch_resource_base.h72 std::vector<Tensor> captured_inputs; member
H A Dbatch_resource_base.cc721 const auto& captured_inputs = in ProcessFuncBatch() local
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dload.py116 def _call_flat(self, args, captured_inputs, cancellation_manager=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dfunction.py1756 def _call_flat(self, args, captured_inputs, cancellation_manager=None): argument
2032 def captured_inputs(self): member in ConcreteFunction
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dfunction.py388 def captured_inputs(self): member in _DefinedFunction