Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc64 assignment_(std::move(assignment)), in CpuExecutable()
66 if (assignment_) { in CpuExecutable()
67 buffer_assignment_.reset(new BufferAssignmentProto(assignment_->ToProto())); in CpuExecutable()
131 assignment_->Allocations().size()); in CreateBufferTable()
132 VLOG(3) << "Allocating " << assignment_->Allocations().size() in CreateBufferTable()
134 for (BufferAllocation::Index i = 0; i < assignment_->Allocations().size(); in CreateBufferTable()
136 const BufferAllocation& allocation = assignment_->GetAllocation(i); in CreateBufferTable()
143 assignment_->GetUniqueTopLevelOutputSlice()); in CreateBufferTable()
186 assignment_->GetUniqueTopLevelOutputSlice()); in ExecuteComputeFunction()
255 this->assignment_->GetUniqueSlice(src, value_source->index())); in CreateResultShapedBuffer()
[all …]
Dcpu_executable.h82 const BufferAssignment& buffer_assignment() const { return *assignment_; } in buffer_assignment()
132 const std::unique_ptr<const BufferAssignment> assignment_; variable
Dir_emitter.cc98 : assignment_(assignment), in IrEmitter()
170 assignment_.GetUniqueTopLevelSlice(computation->root_instruction())); in EmitComputation()
175 assignment_.GetUniqueTopLevelSlice(param)); in EmitComputation()
252 for (const BufferAllocation& allocation : assignment_.Allocations()) { in EmitConstantGlobals()
402 assignment_.GetUniqueSlice(infeed, {0})); in HandleInfeed()
405 assignment_.GetUniqueSlice(infeed, {1})); in HandleInfeed()
420 assignment_.GetUniqueSlice(infeed, {0, i})); in HandleInfeed()
1086 assignment_.GetUniqueSlice(crs, {i})); in HandleAllReduceSingleReplica()
1148 assignment_.GetUniqueSlice(crs, {i})); in HandleAllReduceMultipleReplica()
1158 assignment_.GetUniqueSlice(crs->operand(0), {})); in HandleAllReduceMultipleReplica()
[all …]
Dir_emitter.h181 return assignment_.GetUniqueSlice(&hlo, index).ConsumeValueOrDie();
432 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 auto unique_slice = assignment_.GetUniqueSlice(&hlo, index); in AddAliasingInformationToIrArray()
69 assignment_, hlo); in AddAliasingInformationToIrArray()
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddriver.cc232 explicit BufferTable(BufferAssignment assignment) : assignment_(assignment) { in BufferTable()
245 int num_buffers = assignment_.buffers_size.size(); in ~BufferTable()
253 BufferAssignment assignment_; member in __anon415afa1a0111::BufferTable
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.h46 : assignment_(std::move(assignment)), in LhloDialectEmitter()
252 const xla::BufferAssignment& assignment_; variable
Dmhlo_to_lhlo_with_xla.cc927 assignment_.GetUniqueTopLevelSlice(instr)); in EmitConstant()
1353 assignment_.GetUniqueSlice(instr, shape_index)); in GetOrCreateArrayView()
1446 for (const BufferAllocation& alloc : assignment_.Allocations()) in Initialize()