Searched refs:backward_function_deleter (Results 1 – 1 of 1) sorted by relevance
44 std::function<void(BackwardFunction*)> backward_function_deleter; member120 pair.second.backward_function_deleter(pair.second.backward_function); in ~GradientTape()134 const std::function<void(BackwardFunction*)>& backward_function_deleter);211 const std::function<void(BackwardFunction*)>& backward_function_deleter) { in RecordOperation() argument233 backward_function_deleter}; in RecordOperation()269 op_it->second.backward_function_deleter(op_it->second.backward_function); in DeleteTrace()375 op_pair.second.backward_function_deleter( in PrepareBackprop()496 pair.second.backward_function_deleter(pair.second.backward_function); in ComputeGradient()563 trace.backward_function_deleter(trace.backward_function); in ComputeGradient()572 trace.backward_function_deleter(trace.backward_function); in ComputeGradient()