Searched defs:TensorArgs (Results 1 – 2 of 2) sorted by relevance
117 struct TensorArgs { struct136 TensorArg& lookup(const SavedVariable& sv) { in lookup()142 TensorArg& add(const at::Tensor& tensor) { in add()146 TensorArg& add(const SavedVariable& sv, const std::shared_ptr<Node>& node) { in add()157 std::vector<at::Tensor> inputs;163 std::unordered_map<const SavedVariable*, TensorArg*> _saved_variables;164 TensorArg _undefined;165 uint32_t _next_id = 1; // id=0 used by _undefined
22 using TensorArgs = std::vector<at::Tensor>; variable