Searched refs:buffer_assn (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dump.cc | 185 const BufferAssignment* buffer_assn, in DumpHloModuleImpl() argument
197 buffer_assn ? MakeHloProto(module, *buffer_assn) : MakeHloProto(module); in DumpHloModuleImpl()
273 const BufferAssignment& buffer_assn, in DumpHloModuleIfEnabled() argument
277 DumpHloModuleImpl(module, &buffer_assn, /*profile=*/nullptr, name, opts); in DumpHloModuleIfEnabled()
|
| D | dump.h | 57 const BufferAssignment& buffer_assn,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1502 const BufferAssignment& buffer_assn) { in GetHloBufferSlices() argument
1514 auto slice = buffer_assn.GetUniqueSlice(instr, index); in GetHloBufferSlices()
1526 auto slice = buffer_assn.GetUniqueSlice(parent, {}); in GetHloBufferSlices()
1540 auto slice = buffer_assn.GetUniqueSlice(parent, {}); in GetHloBufferSlices()
1559 auto slice = buffer_assn.GetUniqueSlice(instr, new_index); in GetHloBufferSlices()
1599 const BufferAssignment& buffer_assn = in BuildKernelThunk() local
1604 hlo_slices = GetHloBufferSlices(inst, buffer_assn); in BuildKernelThunk()
1616 for (const BufferAllocation& alloc : buffer_assn.Allocations()) { in BuildKernelThunk()
|