Searched defs:inputs_mapping (Results 1 – 1 of 1) sorted by relevance
30 std::unordered_map<at::TensorImpl*, size_t> inputs_mapping, in _process_forward_mode_AD()257 const std::unordered_map<at::TensorImpl*, size_t>& inputs_mapping, in _process_backward_mode_ad()447 std::unordered_map<at::TensorImpl*, size_t> inputs_mapping; in _wrap_outputs() local