Searched refs:backward_function_getter (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 161 const std::function<BackwardFunction*()>& backward_function_getter,
287 const std::function<BackwardFunction*()>& backward_function_getter,
332 const std::function<BackwardFunction*()>& backward_function_getter,
410 const std::function<BackwardFunction*()>& backward_function_getter, in RecordOperation() argument
432 op_type, std::move(tensors), std::move(ids), backward_function_getter(), in RecordOperation()
926 const std::function<BackwardFunction*()>& backward_function_getter, in ForwardpropFromTape() argument
981 backward_function(backward_function_getter(), in ForwardpropFromTape()
1028 const std::function<BackwardFunction*()>& backward_function_getter, in Accumulate() argument
1043 backward_function_getter, backward_function_deleter); in Accumulate()
1090 op_type, output_tensors, backward_function_getter, in Accumulate()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 2296 const std::function<PyBackwardFunction*()>& backward_function_getter, argument
2305 backward_function_getter,
2316 const std::function<PyBackwardFunction*()>& backward_function_getter, argument
2382 forward_function, backward_function_getter, backward_function_killer);
2517 const std::function<PyBackwardFunction*()>& backward_function_getter, argument
2534 input_dtypes, backward_function_getter, backward_function_killer,
2540 backward_function_getter, backward_function_killer,
2561 std::function<PyBackwardFunction*()> backward_function_getter( local
2580 backward_function_getter, backward_function_killer,
2594 backward_function_getter, backward_function_killer,
[all …]
|