Searched refs:buffer_assn (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dump.cc | 220 const BufferAssignment* buffer_assn, in DumpHloModuleImpl() argument
232 if (buffer_assn) { in DumpHloModuleImpl()
234 StrCat(filename, "-buffer-assignment.txt"), buffer_assn->ToString(), in DumpHloModuleImpl()
241 buffer_assn ? MakeHloProto(module, *buffer_assn) : MakeHloProto(module); in DumpHloModuleImpl()
417 const BufferAssignment& buffer_assn, in DumpHloModuleIfEnabled() argument
421 DumpHloModuleImpl(module, &buffer_assn, /*profile=*/nullptr, in DumpHloModuleIfEnabled()
|
| D | dump.h | 71 const BufferAssignment& buffer_assn,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 3241 const HloInstruction* hlo, const BufferAssignment& buffer_assn) { in GetHloBufferSlices() argument
3253 auto slice = buffer_assn.GetUniqueSlice(instr, index); in GetHloBufferSlices()
3265 auto slice = buffer_assn.GetUniqueSlice(parent, {}); in GetHloBufferSlices()
3279 auto slice = buffer_assn.GetUniqueSlice(parent, {}); in GetHloBufferSlices()
3298 auto slice = buffer_assn.GetUniqueSlice(instr, new_index); in GetHloBufferSlices()
|