Searched defs:forward_fn (Results 1 – 7 of 7) sorted by relevance
40 : forward_fn(std::move(forward)), in forward_fn() function47 std::function<Tensor(const Tensor&, int64_t)> forward_fn; member
11 using forward_fn = void (*)( variable
80 using forward_fn = void (*)( variable
11 using forward_fn = void (*)(const Tensor&, const Tensor&); variable
282 def forward_fn(x, beta, gamma): function
1346 THPObjectPtr forward_fn(PyObject_GetAttrString(cls, "forward")); in THPFunction_apply() local
536 def apply_chunking_to_forward(forward_fn, *input_tensors): argument