Searched refs:NewLogicalBuffer (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | logical_buffer_analysis.cc | 91 void LogicalBufferAnalysis::NewLogicalBuffer(HloInstruction* instruction, in NewLogicalBuffer() function in xla::LogicalBufferAnalysis
107 NewLogicalBuffer(hlo_instruction, index); in DefaultAction()
128 NewLogicalBuffer(copy, /*index=*/{}); in HandleCopy()
148 NewLogicalBuffer(recv_done, /*index=*/{}); in HandleRecvDone()
149 NewLogicalBuffer(recv_done, /*index=*/{1}); in HandleRecvDone()
158 NewLogicalBuffer(send, /*index=*/{}); in HandleSend()
159 NewLogicalBuffer(send, /*index=*/{1}); in HandleSend()
160 NewLogicalBuffer(send, /*index=*/{2}); in HandleSend()
167 NewLogicalBuffer(copy_start, /*index=*/{}); in HandleCopyStart()
168 NewLogicalBuffer(copy_start, /*index=*/{0}); in HandleCopyStart()
[all …]
|
| D | logical_buffer_analysis.h | 57 void NewLogicalBuffer(HloInstruction* instruction, const ShapeIndex& index);
|