Searched defs:buffer_assignment (Results 1 - 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | ir_emitter_context.h | 20 #include "tensorflow/compiler/xla/service/buffer_assignment.h" 34 const BufferAssignment* buffer_assignment, 38 buffer_assignment_(buffer_assignment), 47 const BufferAssignment& buffer_assignment() const { function in class:xla::gpu::IrEmitterContext 33 IrEmitterContext(const HloModule* hlo_module, const BufferAssignment* buffer_assignment, const perftools::gputools::DeviceDescription* device_desc, llvm::Module* llvm_module) argument
|
H A D | buffer_allocations.cc | 42 const BufferAssignment& buffer_assignment, int device_ordinal, 44 const int64 num_buffers = buffer_assignment.Allocations().size(); 67 const BufferAllocation& allocation = buffer_assignment.GetAllocation(i); 114 const BufferAssignment& buffer_assignment) { 116 const int64 num_buffers = buffer_assignment.Allocations().size(); 118 const BufferAllocation& allocation = buffer_assignment.GetAllocation(i); 41 Build( const BufferAssignment& buffer_assignment, int device_ordinal, DeviceMemoryAllocator* memory_allocator) argument 112 TearDown( const std::set<se::DeviceMemoryBase>& live_addresses, const BufferAssignment& buffer_assignment) argument
|
H A D | hlo_to_ir_bindings.h | 24 #include "tensorflow/compiler/xla/service/buffer_assignment.h" 38 const BufferAssignment* buffer_assignment, 41 : buffer_assignment_(buffer_assignment), 37 HloToIrBindings(const HloModule& module, const BufferAssignment* buffer_assignment, llvm::IRBuilder<>* ir_builder, llvm::Module* llvm_module, bool is_nested) argument
|
H A D | hlo_to_ir_bindings.cc | 178 const BufferAssignment* buffer_assignment) { 193 buffer_assignment->HaveDisjointSlices(&hlo, &consumer); 176 BuffersInvariantWithinConsumer( const HloInstruction& hlo, const HloInstruction& consumer, const BufferAssignment* buffer_assignment) argument
|
H A D | ir_emitter_unnested.cc | 31 #include "tensorflow/compiler/xla/service/buffer_assignment.h" 193 bool ImplementedAsHostToDeviceMemcpy(const BufferAssignment& buffer_assignment, argument 206 GetKnownAtRuntimeSlice(&hlo, {}, buffer_assignment).has_value(); 210 const BufferAssignment& buffer_assignment, const HloInstruction& hlo) { 220 GetKnownAtRuntimeSlice(&hlo, {}, buffer_assignment).has_value() && 221 GetKnownAtRuntimeSlice(hlo.operand(0), {}, buffer_assignment) 360 const auto& assn = ir_emitter_context_->buffer_assignment(); 390 const auto& assn = ir_emitter_context_->buffer_assignment(); 412 const auto& assn = ir_emitter_context_->buffer_assignment(); 563 fusion, ir_emitter_context_->buffer_assignment())) { 209 ImplementedAsDeviceToDeviceMemcpy( const BufferAssignment& buffer_assignment, const HloInstruction& hlo) argument 2301 CheckHloBuffersShareAllocation( const HloInstruction* a, const HloInstruction* b, const ShapeIndex& index, const BufferAssignment& buffer_assignment) argument 2321 CheckWhileBuffersShareAllocation( const HloInstruction* xla_while, const BufferAssignment& buffer_assignment) argument 2352 CheckConditionalBuffersShareAllocation( const HloInstruction* conditional, const BufferAssignment& buffer_assignment) argument [all...] |
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
H A D | cpu_executable.h | 25 #include "tensorflow/compiler/xla/service/buffer_assignment.h" 90 const BufferAssignment& buffer_assignment() const { return *assignment_; } function in class:xla::cpu::CpuExecutable
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
H A D | xla_jit_compiled_cpu_function.cc | 54 const xla::BufferAssignment& buffer_assignment) { 56 buffer_assignment.Allocations(); 74 const xla::BufferAssignment& buffer_assignment) { 76 buffer_assignment.GetUniqueTopLevelOutputSlice()); 152 const xla::BufferAssignment& buffer_assignment = local 153 cpu_executable->buffer_assignment(); 159 ComputeTempSizes(buffer_assignment)); 161 ComputeResultIndex(buffer_assignment)); 53 ComputeTempSizes( const xla::BufferAssignment& buffer_assignment) argument 73 ComputeResultIndex( const xla::BufferAssignment& buffer_assignment) argument
|
Completed in 188 milliseconds