| /aosp_15_r20/external/pytorch/torch/autograd/ |
| H A D | functional.py | 69 def _grad_preprocess(inputs, create_graph, need_graph): argument
92 def _grad_postprocess(inputs, create_graph): argument
172 create_graph=False, argument
206 def _fill_in_zeros(grads, refs, strict, create_graph, stage): argument
264 def vjp(func, inputs, v=None, create_graph=False, strict=False): argument
359 def jvp(func, inputs, v=None, create_graph=False, strict=False): argument
578 create_graph=False, argument
835 create_graph=False, argument
970 def vhp(func, inputs, v=None, create_graph=False, strict=False): argument
1071 def hvp(func, inputs, v=None, create_graph=False, strict=False): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | autograd.cpp | 94 bool create_graph, in run_backward()
168 bool create_graph, in backward()
189 bool create_graph, in grad()
|
| H A D | python_engine.cpp | 127 bool create_graph, in execute()
192 unsigned char create_graph = 0; in THPEngine_run_backward() local
|
| H A D | engine.cpp | 1192 bool create_graph, in execute()
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | eager_transforms.py | 166 outputs, inputs, grad_outputs=None, retain_graph=False, create_graph=True argument
425 def wrapper(cotangents, retain_graph=True, create_graph=None): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | register_prim_ops_fulljit.cpp | 231 bool create_graph = pop(stack).toBool(); in __anon1bf66a481902() local
270 bool create_graph = pop(stack).toBool(); in __anon1bf66a481a02() local
|
| H A D | register_prim_ops.cpp | 1131 bool create_graph = pop(stack).toBool(); in __anonbfe5918f4602() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | VariableMethodStubs.cpp | 27 …ts, const std::optional<Tensor>& gradient_opt, std::optional<bool> keep_graph, bool create_graph) { in _backward()
|
| /aosp_15_r20/system/extras/simpleperf/scripts/purgatorio/ |
| H A D | purgatorio.py | 50 def create_graph(args, source, data_range): function
|
| /aosp_15_r20/external/pytorch/torch/utils/viz/ |
| H A D | _cycles.py | 263 def create_graph(objects, *, context=None, filter=None): function
|
| /aosp_15_r20/external/pytorch/test/cpp/lazy/ |
| H A D | test_lazy_ops_util.cpp | 177 bool create_graph = d != derivative_level; in TestBackward() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_decomp.py | 103 outputs, inputs, grad_outputs=None, retain_graph=False, create_graph=True argument
|
| H A D | test_autograd.py | 805 def compute_grad(create_graph): argument
827 create_graph, argument
|
| H A D | test_nn.py | 86 def _backward(self, module, input: _TensorOrTensors, output, grad_output, create_graph=False): argument
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _tensor.py | 526 self, gradient=None, retain_graph=None, create_graph=False, inputs=None argument
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_ops.py | 74 outputs, inputs, grad_outputs=None, retain_graph=False, create_graph=True argument
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_cudagraph_trees.py | 1218 def compute_grad(grad_output, create_graph): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | mark_for_compilation_pass_test.cc | 1765 auto create_graph = in TEST() local
|