Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/autograd/functions/
H A Dtensor.cpp24 if (task_should_compute_output(0)) { in apply()
27 if (task_should_compute_output(1)) { in apply()
120 if (task_should_compute_output(0)) { in apply_impl()
136 it->second.should_execute() == task_should_compute_output(0)); in apply_impl()
156 if (task_should_compute_output(i)) { in apply_impl()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dfunction.h419 bool task_should_compute_output(size_t output_edge_index) const { in task_should_compute_output() function
437 bool task_should_compute_output( in task_should_compute_output() function
441 if (task_should_compute_output(i)) in task_should_compute_output()
H A Dcustom_function.cpp543 return ptr->task_should_compute_output(output_edge_index); in needs_input_grad()
550 return ptr->task_should_compute_output(idxs); in needs_input_grad()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dgraph_executor.cpp382 if (task_should_compute_output(i)) { in produceOutput()