Cross Reference: /external/tensorflow/tensorflow/compiler/xla/service/buffer

Lines Matching refs:buffers
82   // computation. These buffers have lifetimes which may be longer than the
109   // Returns the color of the allocation. Only logical buffers with a matching
118   // Access to the logical buffers assigned to this allocation, and their
186     // Parameters do not need temporary buffers.
191            // Thread-local buffers are allocated using `alloca`s.
257   // Mapping from the set of buffers assigned to this allocation to their
275   // Returns the total size allocation holding all temporary buffers.
336   // Returns true if the top-level buffers of hlo_a and hlo_b are the same.
345   // buffers are all different from hlo_b's buffers.
418   // Combines allocations of temporary buffers into one big BufferAllocation.
427   // The total size of all temporary buffers.
456   // are allowed to be reused as outbut buffers by the client code.
477   // Assigns buffers to the instructions in the given computation. "assignment"
479   // true, then all assigned buffers have the is_thread_local flag set to
516   // Colocated buffers are logical buffers from different computations which
517   // alias. Explicitly handling these colocated buffers is necessary because
530   // For each buffer set in 'colocated_buffer_sets', assigns all buffers in the
538   // Adds the 'colocated_set' of buffers to 'colocated_buffer_sets', maintaining
545   // the logical buffers that would be assigned to the same physical buffer),
546   // try to merge the sets if the buffers can be shared. Returns the merged set.
552   // Split a set of buffers into several sets, each of which contains buffers
558       const tensorflow::gtl::FlatSet<const LogicalBuffer*>& buffers);
560   // If true, buffer assignments assumes that input parameter buffers and output
561   // buffers can be shared if their sizes match.
564   // Functor used to assign colors to newly allocated logical buffers.