Searched defs:buffer_assignment (Results 1 - 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_context.h20 #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 Dbuffer_allocations.cc42 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 Dhlo_to_ir_bindings.h24 #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 Dhlo_to_ir_bindings.cc178 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 Dir_emitter_unnested.cc31 #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 Dcpu_executable.h25 #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 Dxla_jit_compiled_cpu_function.cc54 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