Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dupdate_differentiable_graph_requires_grad.cpp8 static void UpdateDifferentiableGraphRequiresGrad( in UpdateDifferentiableGraphRequiresGrad() function
26 UpdateDifferentiableGraphRequiresGrad(b, new_requires_grad); in UpdateDifferentiableGraphRequiresGrad()
31 void UpdateDifferentiableGraphRequiresGrad( in UpdateDifferentiableGraphRequiresGrad() function
34 UpdateDifferentiableGraphRequiresGrad( in UpdateDifferentiableGraphRequiresGrad()
H A Dupdate_differentiable_graph_requires_grad.h14 TORCH_API void UpdateDifferentiableGraphRequiresGrad(
H A Dinline_autodiff_subgraphs.cpp70 UpdateDifferentiableGraphRequiresGrad(subgraph, std::nullopt); in scanNode()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dprofiling_graph_executor_impl.cpp322 UpdateDifferentiableGraphRequiresGrad( in guardDifferentiableGraph()
516 UpdateDifferentiableGraphRequiresGrad(gradient.f, false); in runProfilingOptimizations()
H A Dautodiff.cpp863 UpdateDifferentiableGraphRequiresGrad(grad_desc.f, false); in differentiate()