Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dtuple_points_to_analysis.cc155 logical_buffer_analysis_->num_logical_buffers()); in Analyze()
211 &logical_buffer_analysis_->GetBuffer(hlo_instruction, index)); in DefaultAction()
262 logical_buffer_analysis_->GetBuffer(copy, /*index=*/{}), in HandleCopy()
296 logical_buffer_analysis_->GetBuffer(recv_done, /*index=*/{}), in HandleRecvDone()
299 logical_buffer_analysis_->GetBuffer(recv_done, /*index=*/{1}), in HandleRecvDone()
338 &logical_buffer_analysis_->GetBuffer(async_start, target_index)); in HandleAsyncStart()
398 &logical_buffer_analysis_->GetBuffer(copy_start, target_index)); in HandleCopyStart()
440 &logical_buffer_analysis_->GetBuffer(send, ShapeIndex({}))); in HandleSend()
445 &logical_buffer_analysis_->GetBuffer(send, ShapeIndex({1}))); in HandleSend()
449 &logical_buffer_analysis_->GetBuffer(send, ShapeIndex({2}))); in HandleSend()
[all …]
H A Dtuple_points_to_analysis.h211 return logical_buffer_analysis_->num_logical_buffers(); in num_logical_buffers()
222 return logical_buffer_analysis_->GetBuffer(id); in logical_buffer()
278 logical_buffer_analysis_(std::move(logical_buffer_analysis)) {} in TuplePointsToAnalysis()
349 const std::unique_ptr<LogicalBufferAnalysis> logical_buffer_analysis_; variable