Searched refs:BackwardFunction (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 39 template <typename BackwardFunction, typename TapeTensor>
46 BackwardFunction* backward_function;
50 std::function<void(BackwardFunction*)> backward_function_deleter;
59 template <typename BackwardFunction, typename TapeTensor>
61 std::unordered_map<int64, OpTapeEntry<BackwardFunction, TapeTensor>>;
82 template <typename Gradient, typename BackwardFunction, typename TapeTensor>
100 const string& op_type, BackwardFunction* backward_function,
125 template <typename Gradient, typename BackwardFunction, typename TapeTensor>
161 const std::function<BackwardFunction*()>& backward_function_getter,
162 const std::function<void(BackwardFunction*)>& backward_function_deleter);
[all …]
|