Searched refs:ConsumeArg (Results 1 – 6 of 6) sorted by relevance
65 frame->ConsumeArg(index_, &val); in Compute()
466 void ConsumeArg(int index, Tensor* val) override { in ConsumeArg() function in tensorflow::__anona00ae9de0111::WhileOp::BodyFuncCallFrame
366 args.call_frame->ConsumeArg(i, first_input.val.get()); in Run()
437 void ConsumeArg(int index, Tensor* val) override { *val = std::move(*arg_); } in ConsumeArg() function in tensorflow::__anon153a2bb70111::ConsumeArgumentCallFrame
303 virtual void ConsumeArg(int index, Tensor* val) { in ConsumeArg() function
331 void ConsumeArg(int index, Tensor* val) override { in ConsumeArg() function in tensorflow::data::__anona21e91bd0111::OwnedArgsCallFrame