Home | Sort by relevance Sort by last modified time |
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ | |
buffer_allocations.h | 92 buffer_assignment_(buffer_assignment) {} 108 const BufferAssignment* buffer_assignment_; member in class:xla::gpu::BufferAllocations |
hlo_to_ir_bindings.h | 42 : buffer_assignment_(buffer_assignment), 46 alias_analysis_(module, *buffer_assignment_, &b_->getContext()) {} 103 const BufferAssignment* buffer_assignment_; member in class:xla::gpu::HloToIrBindings |
ir_emitter_context.h | 38 buffer_assignment_(buffer_assignment), 48 return *buffer_assignment_; 58 const BufferAssignment* buffer_assignment_; member in class:xla::gpu::IrEmitterContext |