Searched defs:CustomOpAutogradFunction (Results 1 – 1 of 1) sorted by relevance
26 struct CustomOpAutogradFunction : public torch::autograd::Function<CustomOpAutogradFunction> { struct27 static torch::Tensor forward( in forward()63 return CustomOpAutogradFunction::apply(var1, mul, var2, var3); in custom_op_with_autograd() argument