Home
last modified time | relevance | path

Searched refs:buffers_needed (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc3351 std::unordered_set<const BufferAllocation*> buffers_needed; in BuildKernelThunkFromBufferSlices() local
3353 buffers_needed.insert(slice->buffer_slice.allocation()); in BuildKernelThunkFromBufferSlices()
3365 buffers_needed.insert(*temp_buffer); in BuildKernelThunkFromBufferSlices()
3371 absl::c_copy_if(buffers_needed, std::back_inserter(non_constant_buffers), in BuildKernelThunkFromBufferSlices()