| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| alias_analysis.h | 36 : module_(module), assignment_(assignment), context_(context) {}
68 const BufferAssignment& assignment_; member in class:xla::llvm_ir::AliasAnalysis
|
| alias_analysis.cc | 44 assignment_.GetAllSlices(&hlo, index);
69 assignment_, hlo);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_executable.cc | 60 assignment_(std::move(assignment)) {
81 assignment_->Allocations().size());
83 assignment_->Allocations().size());
84 VLOG(3) << "Allocating " << assignment_->Allocations().size()
86 for (BufferAllocation::Index i = 0; i < assignment_->Allocations().size();
88 auto& allocation = assignment_->GetAllocation(i);
130 assignment_->GetUniqueTopLevelOutputSlice());
169 assignment_->GetUniqueTopLevelOutputSlice());
236 this->assignment_->GetUniqueSlice(src, buffer_source->index()));
357 return assignment_->points_to_analysis().GetPointsToSet
[all...] |
| cpu_executable.h | 86 const BufferAssignment& buffer_assignment() const { return *assignment_; }
139 const std::unique_ptr<const BufferAssignment> assignment_; member in class:xla::cpu::CpuExecutable
|
| ir_emitter.h | 193 return assignment_.GetUniqueSlice(&hlo, index).ConsumeValueOrDie();
419 const BufferAssignment& assignment_; member in namespace:xla::cpu
|
| ir_emitter.cc | 90 : assignment_(assignment),
125 assignment_.GetUniqueTopLevelSlice(computation->root_instruction()));
130 assignment_.GetUniqueTopLevelSlice(param));
192 for (const BufferAllocation& allocation : assignment_.Allocations()) {
338 assignment_.GetUniqueSlice(infeed, {0}));
341 assignment_.GetUniqueSlice(infeed, {1}));
356 assignment_.GetUniqueSlice(infeed, {0, i}));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gpu_executable.cc | 64 assignment_(std::move(assignment)) {}
205 for (BufferAllocation::Index i = 0; i < assignment_->Allocations().size();
207 const BufferAllocation& allocation = assignment_->GetAllocation(i);
251 for (BufferAllocation::Index i = 0; i < assignment_->Allocations().size();
253 const BufferAllocation& allocation = assignment_->GetAllocation(i);
280 assignment_.get(), executor->device_ordinal(), memory_allocator));
312 this->assignment_->GetUniqueSlice(src_hlo, sources[0]->index()));
358 return assignment_->points_to_analysis().GetPointsToSet(
|
| gpu_executable.h | 138 const std::unique_ptr<const BufferAssignment> assignment_; member in class:xla::gpu::GpuExecutable
|
