Home
last modified time | relevance | path

Searched defs:input_nr (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dfunction.h200 uint32_t input_nr = input_metadata_.size(); in add_input_metadata() local
208 uint32_t input_nr = input_metadata_.size(); in add_input_metadata() local
215 uint32_t input_nr = input_metadata_.size(); in add_input_metadata() local
731 const auto input_nr = function->add_input_metadata(variable); in create_gradient_edge() local
H A Dedge.h38 uint32_t input_nr; member
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Donnx.cpp463 Py_ssize_t input_nr = 0; in NodeToONNX() local
550 Py_ssize_t input_nr = 0; in NodeToONNX() local
/aosp_15_r20/external/pytorch/torch/csrc/dynamo/
H A Dpython_compiled_autograd.cpp584 int input_nr = graph_output.first; in _compiled_autograd_impl() local