Searched refs:backward_function_getter (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | pywrap_tfe_src.cc | 2368 const std::function<PyBackwardFunction*()>& backward_function_getter, argument
2377 backward_function_getter,
2388 const std::function<PyBackwardFunction*()>& backward_function_getter, argument
2454 forward_function, backward_function_getter, backward_function_killer);
2589 const std::function<PyBackwardFunction*()>& backward_function_getter, argument
2606 input_dtypes, backward_function_getter, backward_function_killer,
2612 backward_function_getter, backward_function_killer,
2633 std::function<PyBackwardFunction*()> backward_function_getter( local
2652 backward_function_getter, backward_function_killer,
2666 backward_function_getter, backward_function_killer,
[all …]
|