Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/autograd/functions/
Dinit.cpp40 UndefinedGrad* operator()(PyObject* args) { in operator ()()
45 return new UndefinedGrad(); in operator ()()
146 addClass<UndefinedGrad, UndefinedGradCtor>( in THPAutograd_initFunctions()
Dbasic_ops.h57 struct TORCH_API UndefinedGrad : public Node { struct
58 UndefinedGrad() { in UndefinedGrad() function
Dbasic_ops.cpp42 auto UndefinedGrad::apply(variable_list&& inputs) -> variable_list { in apply()
/external/pytorch/torch/_C/
D_functions.pyi5 class UndefinedGrad:
/external/pytorch/test/
Dtest_autograd_fallback.py360 w = torch._C._functions.UndefinedGrad()(w)
361 z = torch._C._functions.UndefinedGrad()(z)
Dtest_autograd.py353 torch._C._functions.UndefinedGrad()(MyFunction.apply(x)).backward()
368 torch._C._functions.UndefinedGrad()(MyFunction.apply(x)).backward()
9837 t3 = torch._C._functions.UndefinedGrad()(t3)
/external/pytorch/test/cpp/api/
Dautograd.cpp38 auto y = torch::autograd::UndefinedGrad().apply({x}); in TEST()
46 auto y = torch::autograd::UndefinedGrad().apply({x}); in TEST()
527 UndefinedGrad().apply({MyFunction::apply(x)})[0].backward(); in TEST()
546 UndefinedGrad().apply({MyFunction::apply(x)})[0].backward(); in TEST()
/external/pytorch/torch/autograd/
Dgradcheck.py1336 torch._C._functions.UndefinedGrad()(o)
1349 torch._C._functions.UndefinedGrad()(o)
/external/pytorch/torch/testing/_internal/distributed/
Ddistributed_test.py8677 torch._C._functions.UndefinedGrad()(out).backward()
8678 torch._C._functions.UndefinedGrad()(local_out).backward()