Searched refs:BackwardFunction (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 33 template <typename BackwardFunction, typename TapeTensor>
40 BackwardFunction* backward_function;
44 std::function<void(BackwardFunction*)> backward_function_deleter;
53 template <typename BackwardFunction, typename TapeTensor>
54 using OpTape = gtl::FlatMap<int64, OpTapeEntry<BackwardFunction, TapeTensor>>;
75 template <typename Gradient, typename BackwardFunction, typename TapeTensor>
96 BackwardFunction* backward_function,
110 template <typename Gradient, typename BackwardFunction, typename TapeTensor>
133 const std::function<BackwardFunction*()>& backward_function_getter,
134 const std::function<void(BackwardFunction*)>& backward_function_deleter);
[all …]
|