Home
last modified time | relevance | path

Searched refs:backward_function_deleter (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dtape.h50 std::function<void(BackwardFunction*)> backward_function_deleter; member
135 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() argument
433 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 …]