Searched refs:op_tape_ (Results 1 – 1 of 1) sorted by relevance
119 for (const auto& pair : op_tape_) { in ~GradientTape()154 OpTape<BackwardFunction, TapeTensor> op_tape_; variable231 op_tape_[op_id] = OpTapeEntry<BackwardFunction, TapeTensor>{ in RecordOperation()258 auto op_it = op_tape_.find(op_id); in DeleteTrace()259 CHECK(op_it != op_tape_.end()); in DeleteTrace()270 op_tape_.erase(op_it); in DeleteTrace()485 target_tensor_ids, tensor_tape_, &op_tape_, sources_set, persistent_); in ComputeGradient()