Searched refs:backward_tape (Results 1 – 1 of 1) sorted by relevance
347 GradientTape<Gradient, BackwardFunction, TapeTensor>* backward_tape, in AccumulatorCallState()349 : backward_tape(backward_tape), accumulating(accumulating) {} in AccumulatorCallState()356 GradientTape<Gradient, BackwardFunction, TapeTensor>* backward_tape; member903 if (call_state_.top().backward_tape != nullptr) { in ShouldRecord()906 return call_state_.top().backward_tape->ShouldRecord(tensor_ids, dtypes); in ShouldRecord()941 call_state.backward_tape = tape.get(); in ForwardpropFromTape()943 gtl::MakeCleanup([&call_state] { call_state.backward_tape = nullptr; }); in ForwardpropFromTape()1030 if (call_state_.top().backward_tape != nullptr) { in Accumulate()1041 call_state_.top().backward_tape->RecordOperation( in Accumulate()