Searched refs:check_input_variables (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/functions/ |
| H A D | tensor.cpp | 20 check_input_variables("CopyBackwards", grads, 1, -1, true); in apply()
79 check_input_variables("CopySlices", inputs, 1, -1, true); in apply_impl()
|
| H A D | utils.cpp | 44 void check_input_variables( in check_input_variables() function
|
| H A D | utils.h | 32 TORCH_API void check_input_variables(
|
| H A D | accumulate_grad.cpp | 25 check_input_variables("AccumulateGrad", grads, 1, 0); in apply()
|