Home
last modified time | relevance | path

Searched refs:all_tensors_on_stack (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dautograd_not_implemented_fallback.cpp133 std::vector<const at::Tensor*> all_tensors_on_stack; in basicAutogradNotImplementedFallbackImpl() local
136 all_tensors_on_stack.push_back(&t); in basicAutogradNotImplementedFallbackImpl()
142 new WarnNotImplemented(op_name, all_tensors_on_stack.size()), in basicAutogradNotImplementedFallbackImpl()
144 grad_fn->set_next_edges(collect_next_edges(all_tensors_on_stack)); in basicAutogradNotImplementedFallbackImpl()