Returns true if the whole tensor needs to be cached/buffered in memory.
(self)
| 486 | return self._parameters.use_compact_trace |
| 487 | |
| 488 | def _use_tensor_buffer(self): |
| 489 | """Returns true if the whole tensor needs to be cached/buffered in memory.""" |
| 490 | return (self._parameters.trace_mode == |
| 491 | tensor_tracer_flags.TRACE_MODE_FULL_TENSOR_SUMMARY) |
| 492 | |
| 493 | def _save_tensor_value_to_cache_op(self, cache_idx, updates): |
| 494 | """Returns an op that will save the given updates to an entry in the cache. |