Searched refs:backward_function_deleter (Results 1 – 1 of 1) sorted by relevance
50 std::function<void(BackwardFunction*)> backward_function_deleter; member135 pair.second.backward_function_deleter(pair.second.backward_function); in ~GradientTape()162 const std::function<void(BackwardFunction*)>& backward_function_deleter);288 const std::function<void(BackwardFunction*)>& backward_function_deleter);333 const std::function<void(BackwardFunction*)>& backward_function_deleter,411 const std::function<void(BackwardFunction*)>& backward_function_deleter) { in RecordOperation() argument433 backward_function_deleter}; in RecordOperation()469 op_it->second.backward_function_deleter(op_it->second.backward_function); in DeleteTrace()575 op_pair.second.backward_function_deleter( in PrepareBackprop()703 pair.second.backward_function_deleter(pair.second.backward_function); in ComputeGradient()[all …]