Searched refs:backward_function_getter (Results 1 – 2 of 2) sorted by relevance
133 const std::function<BackwardFunction*()>& backward_function_getter,210 const std::function<BackwardFunction*()>& backward_function_getter, in RecordOperation() argument232 op_type, std::move(tensors), std::move(ids), backward_function_getter(), in RecordOperation()
1579 const std::function<PyBackwardFunction*()>& backward_function_getter, argument1615 input_dtypes, backward_function_getter,