Searched refs:ThreadTapeIsStopped (Results 1 – 1 of 1) sorted by relevance
1849 bool* ThreadTapeIsStopped() { in ThreadTapeIsStopped() function1854 void TFE_Py_TapeSetStopOnThread() { *ThreadTapeIsStopped() = true; } in TFE_Py_TapeSetStopOnThread()1856 void TFE_Py_TapeSetRestartOnThread() { *ThreadTapeIsStopped() = false; } in TFE_Py_TapeSetRestartOnThread()1859 if (*ThreadTapeIsStopped()) { in TFE_Py_TapeSetIsStopped()1890 if (*ThreadTapeIsStopped() || !HasAccumulatorOrTape()) { in TFE_Py_TapeSetIsEmpty()1959 if (*ThreadTapeIsStopped()) {1968 bool CouldBackprop() { return !*ThreadTapeIsStopped() && HasGradientTape(); }1970 bool CouldForwardprop() { return !*ThreadTapeIsStopped() && HasAccumulator(); }2551 if (!HasAccumulatorOrTape() || *ThreadTapeIsStopped()) {2605 if (!HasAccumulator() || *ThreadTapeIsStopped()) {[all …]