Searched defs:output_nr (Results 1 – 10 of 10) sorted by relevance
280 bool is_saved_and_setup_context) { in _process_backward_mode_ad()358 auto output_nr = cdata->add_input_metadata(Node::undefined_input()); in _process_backward_mode_ad() local378 uint32_t output_nr = 0; in _process_backward_mode_ad() local
118 const auto output_nr = input.output_nr(); in run_backward() local
148 auto output_nr = grad_fn->add_input_metadata(var); in rebase_history() local
167 auto output_nr = THPUtils_unpackLong(PyTuple_GetItem(obj, 1)); in parseGradientEdge() local
465 int64_t VariableHooks::output_nr(const at::TensorBase& self) const { in output_nr() function in torch::autograd::VariableHooks
1137 const auto output_nr = in THPVariable_get_output_nr() local
43 int64_t output_nr(const Tensor& self) { in output_nr() function
75 auto output_nr = grad_fn->add_input_metadata(variable); in set_history() local
112 int64_t TensorBase::output_nr() const { in output_nr() function in at::TensorBase
111 int output_nr = 0; in updateOutputTensorTracker() local