Home
last modified time | relevance | path

Searched refs:assignment_ (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc60 assignment_(std::move(assignment)) { in CpuExecutable()
81 assignment_->Allocations().size()); in CreateBufferTable()
83 assignment_->Allocations().size()); in CreateBufferTable()
84 VLOG(3) << "Allocating " << assignment_->Allocations().size() in CreateBufferTable()
86 for (BufferAllocation::Index i = 0; i < assignment_->Allocations().size(); in CreateBufferTable()
88 auto& allocation = assignment_->GetAllocation(i); in CreateBufferTable()
130 assignment_->GetUniqueTopLevelOutputSlice()); in CreateBufferTable()
169 assignment_->GetUniqueTopLevelOutputSlice()); in ExecuteComputeFunction()
236 this->assignment_->GetUniqueSlice(src, buffer_source->index())); in CreateResultShapedBuffer()
357 return assignment_->points_to_analysis().GetPointsToSet( in GetRootPointsToSet()
Dcpu_executable.h86 const BufferAssignment& buffer_assignment() const { return *assignment_; } in buffer_assignment()
139 const std::unique_ptr<const BufferAssignment> assignment_; variable
Dir_emitter.cc90 : assignment_(assignment), in IrEmitter()
125 assignment_.GetUniqueTopLevelSlice(computation->root_instruction())); in EmitComputation()
130 assignment_.GetUniqueTopLevelSlice(param)); in EmitComputation()
192 for (const BufferAllocation& allocation : assignment_.Allocations()) { in EmitConstantGlobals()
338 assignment_.GetUniqueSlice(infeed, {0})); in HandleInfeed()
341 assignment_.GetUniqueSlice(infeed, {1})); in HandleInfeed()
356 assignment_.GetUniqueSlice(infeed, {0, i})); in HandleInfeed()
1335 assignment_.GetUniqueSlice(crs, {i})); in HandleAllReduce()
2070 assignment_)) { in HandleDynamicUpdateSlice()
2151 if (llvm_ir::CanEmitFusedDynamicUpdateSliceInPlace(fusion, assignment_)) { in HandleFusion()
[all …]
Dir_emitter.h193 return assignment_.GetUniqueSlice(&hlo, index).ConsumeValueOrDie();
419 const BufferAssignment& assignment_; variable
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc64 assignment_(std::move(assignment)) {} in GpuExecutable()
205 for (BufferAllocation::Index i = 0; i < assignment_->Allocations().size(); in ResolveConstantGlobals()
207 const BufferAllocation& allocation = assignment_->GetAllocation(i); in ResolveConstantGlobals()
251 for (BufferAllocation::Index i = 0; i < assignment_->Allocations().size(); in ExecuteOnStream()
253 const BufferAllocation& allocation = assignment_->GetAllocation(i); in ExecuteOnStream()
280 assignment_.get(), executor->device_ordinal(), memory_allocator)); in ExecuteOnStream()
312 this->assignment_->GetUniqueSlice(src_hlo, sources[0]->index())); in ExecuteOnStream()
358 return assignment_->points_to_analysis().GetPointsToSet( in GetRootPointsToSet()
Dgpu_executable.h138 const std::unique_ptr<const BufferAssignment> assignment_; variable
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dalias_analysis.h36 : module_(module), assignment_(assignment), context_(context) {} in AliasAnalysis()
68 const BufferAssignment& assignment_; variable
Dalias_analysis.cc44 assignment_.GetAllSlices(&hlo, index); in AddAliasingInformationToIrArray()
69 assignment_, hlo); in AddAliasingInformationToIrArray()