Searched refs:logical_buffer_analysis_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tuple_points_to_analysis.cc | 155 logical_buffer_analysis_->num_logical_buffers()); in Analyze()
210 &logical_buffer_analysis_->GetBuffer(hlo_instruction, index)); in DefaultAction()
261 logical_buffer_analysis_->GetBuffer(copy, /*index=*/{}), in HandleCopy()
295 logical_buffer_analysis_->GetBuffer(recv_done, /*index=*/{}), in HandleRecvDone()
298 logical_buffer_analysis_->GetBuffer(recv_done, /*index=*/{1}), in HandleRecvDone()
332 &logical_buffer_analysis_->GetBuffer(copy_start, target_index)); in HandleCopyStart()
374 &logical_buffer_analysis_->GetBuffer(send, ShapeIndex({}))); in HandleSend()
379 &logical_buffer_analysis_->GetBuffer(send, ShapeIndex({1}))); in HandleSend()
383 &logical_buffer_analysis_->GetBuffer(send, ShapeIndex({2}))); in HandleSend()
410 logical_buffer_analysis_->GetBuffer(tuple, /*index=*/{}), in HandleTuple()
[all …]
|
| D | tuple_points_to_analysis.h | 212 return logical_buffer_analysis_->num_logical_buffers(); in num_logical_buffers()
223 return logical_buffer_analysis_->GetBuffer(id); in logical_buffer()
276 logical_buffer_analysis_(std::move(logical_buffer_analysis)) {} in TuplePointsToAnalysis()
345 const std::unique_ptr<LogicalBufferAnalysis> logical_buffer_analysis_; variable
|