Searched refs:ConsumeArg (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ | ||
| D | function_ops.cc | 61 frame->ConsumeArg(index_, &val); in Compute() |
| D | functional_ops.cc | 466 void ConsumeArg(int index, Tensor* val) override { in ConsumeArg() function in tensorflow::__anonf02e412f0111::WhileOp::BodyFuncCallFrame |
| /external/tensorflow/tensorflow/core/kernels/data/ | ||
| D | single_threaded_executor.cc | 337 args.call_frame->ConsumeArg(i, first_input.val.get()); in Run() |
| D | captured_function.cc | 354 void ConsumeArg(int index, Tensor* val) override { in ConsumeArg() function in tensorflow::data::__anon5a9b84710111::OwnedArgsCallFrame |
| /external/tensorflow/tensorflow/core/framework/ | ||
| D | function.h | 285 virtual void ConsumeArg(int index, Tensor* val) { in ConsumeArg() function |
| /external/tensorflow/tensorflow/core/common_runtime/ | ||
| D | function_test.cc | 454 void ConsumeArg(int index, Tensor* val) override { *val = std::move(*arg_); } in ConsumeArg() function in tensorflow::__anon196ffbc80111::ConsumeArgumentCallFrame |