Searched defs:ConsumeArg (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ | ||
| H A D | function.h | 303 virtual void ConsumeArg(int index, Tensor* val) { in ConsumeArg() function |
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ | ||
| H A D | functional_ops.cc | 466 void ConsumeArg(int index, Tensor* val) override { in ConsumeArg() function in tensorflow::__anona00ae9de0111::WhileOp::BodyFuncCallFrame |
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ | ||
| H A D | captured_function.cc | 331 void ConsumeArg(int index, Tensor* val) override { in ConsumeArg() function in tensorflow::data::__anona21e91bd0111::OwnedArgsCallFrame |
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ | ||
| H A D | function_test.cc | 437 void ConsumeArg(int index, Tensor* val) override { *val = std::move(*arg_); } in ConsumeArg() function in tensorflow::__anon153a2bb70111::ConsumeArgumentCallFrame |