Searched defs:DifferentiableGraphBackward (Results 1 – 1 of 1) sorted by relevance
251 struct DifferentiableGraphBackward : public autograd::Node { struct252 DifferentiableGraphBackward( in DifferentiableGraphBackward() function260 variable_list apply(variable_list&& inputs) override { in apply()320 void capture(const IValue& val, bool is_output) { in capture()324 void addOutputForTensor(const at::Tensor& tensor) { in addOutputForTensor()330 void addOutputForIValue(const IValue& value) { in addOutputForIValue()347 void addInputVariable(Variable output) { in addInputVariable()360 void addInputIValue(const IValue& v) { in addInputIValue()376 void release_variables() override { in release_variables()381 void produceOutput(size_t i, at::Tensor output, variable_list& outputs) { in produceOutput()[all …]