Searched refs:NewLogicalBuffer (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | logical_buffer_analysis.cc | 89 void LogicalBufferAnalysis::NewLogicalBuffer(HloInstruction* instruction, in NewLogicalBuffer() function in xla::LogicalBufferAnalysis
105 NewLogicalBuffer(hlo_instruction, index); in DefaultAction()
126 NewLogicalBuffer(copy, /*index=*/{}); in HandleCopy()
146 NewLogicalBuffer(recv_done, /*index=*/{}); in HandleRecvDone()
147 NewLogicalBuffer(recv_done, /*index=*/{1}); in HandleRecvDone()
156 NewLogicalBuffer(send, /*index=*/{}); in HandleSend()
157 NewLogicalBuffer(send, /*index=*/{1}); in HandleSend()
158 NewLogicalBuffer(send, /*index=*/{2}); in HandleSend()
164 NewLogicalBuffer(tuple, /*index=*/{}); in HandleTuple()
171 NewLogicalBuffer(tuple_select, /*index=*/{}); in HandleTupleSelect()
|
| D | logical_buffer_analysis.h | 56 void NewLogicalBuffer(HloInstruction* instruction, const ShapeIndex& index);
|