HomeSort by relevance Sort by last modified time
    Searched refs:buffer_assignment_ (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter_context.h 38 buffer_assignment_(buffer_assignment),
48 return *buffer_assignment_;
58 const BufferAssignment* buffer_assignment_; member in class:xla::gpu::IrEmitterContext
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
buffer_allocations.h 92 buffer_assignment_(buffer_assignment) {}
108 const BufferAssignment* buffer_assignment_; member in class:xla::gpu::BufferAllocations
buffer_allocations.cc 133 const int64 num_buffers = buffer_assignment_->Allocations().size();
135 const BufferAllocation& allocation = buffer_assignment_->GetAllocation(i);
hlo_to_ir_bindings.cc 76 buffer_assignment_
92 if (!buffer_assignment_->HasTopLevelAllocation(non_io_hlo)) {
104 buffer_assignment_->GetUniqueSlice(non_io_hlo, index);
229 if (BuffersInvariantWithinConsumer(hlo, consumer, buffer_assignment_)) {

Completed in 406 milliseconds