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 | 152 logical_buffer_analysis_->num_logical_buffers()); in Analyze()
207 &logical_buffer_analysis_->GetBuffer(hlo_instruction, index)); in DefaultAction()
258 logical_buffer_analysis_->GetBuffer(copy, /*index=*/{}), in HandleCopy()
292 logical_buffer_analysis_->GetBuffer(recv_done, /*index=*/{}), in HandleRecvDone()
295 logical_buffer_analysis_->GetBuffer(recv_done, /*index=*/{1}), in HandleRecvDone()
324 &logical_buffer_analysis_->GetBuffer(send, ShapeIndex({}))); in HandleSend()
329 &logical_buffer_analysis_->GetBuffer(send, ShapeIndex({1}))); in HandleSend()
333 &logical_buffer_analysis_->GetBuffer(send, ShapeIndex({2}))); in HandleSend()
360 logical_buffer_analysis_->GetBuffer(tuple, /*index=*/{}), in HandleTuple()
423 logical_buffer_analysis_->GetBuffer(tuple_select, /*index=*/{}), in HandleTupleSelect()
[all …]
|
| D | tuple_points_to_analysis.h | 211 return logical_buffer_analysis_->num_logical_buffers(); in num_logical_buffers()
222 return logical_buffer_analysis_->GetBuffer(id); in logical_buffer()
281 logical_buffer_analysis_(std::move(logical_buffer_analysis)) {} in TuplePointsToAnalysis()
350 const std::unique_ptr<LogicalBufferAnalysis> logical_buffer_analysis_; variable
|