| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment.h | 275 friend class BufferAssignment; variable
351 class BufferAssignment {
473 BufferAssignment(const HloModule* module, in BufferAssignment() function
532 TF_DISALLOW_COPY_AND_ASSIGN(BufferAssignment);
540 const BufferAssignment& assignment, const BufferAllocation& alloc,
548 static StatusOr<std::unique_ptr<BufferAssignment>> Run(
567 StatusOr<std::unique_ptr<BufferAssignment>> CreateAssignment(
583 BufferAssignment* assignment);
594 bool run_whole_module_heap_simulation, BufferAssignment* assignment);
599 BufferAssignment* assignment,
[all …]
|
| D | buffer_assignment.cc | 336 const PointsToSet& BufferAssignment::GetPointsToSet( in GetPointsToSet()
341 bool BufferAssignment::HasAllocation(const LogicalBuffer& buffer) const { in HasAllocation()
346 const BufferAllocation& BufferAssignment::GetAssignedAllocation( in GetAssignedAllocation()
352 BufferAllocation* BufferAssignment::GetMutableAssignedAllocation( in GetMutableAssignedAllocation()
357 std::set<BufferAllocation::Slice> BufferAssignment::GetAllSlices( in GetAllSlices()
368 const BufferAllocation& BufferAssignment::GetAllocation( in GetAllocation()
375 const BufferAllocation* BufferAssignment::GetInstructionAllocation( in GetInstructionAllocation()
389 BufferAllocation* BufferAssignment::GetMutableAllocation( in GetMutableAllocation()
394 bool BufferAssignment::HasAllocationAt(const HloInstruction* instruction, in HasAllocationAt()
405 bool BufferAssignment::HasTopLevelAllocation( in HasTopLevelAllocation()
[all …]
|
| D | dump.h | 32 class BufferAssignment; variable
57 const BufferAssignment& buffer_assn,
|
| D | hlo_proto_util.h | 32 const BufferAssignment& assignment);
|
| D | hlo_proto_util.cc | 26 const BufferAssignment& assignment) { in MakeHloProto()
|
| D | buffer_assignment_test.cc | 89 std::unique_ptr<BufferAssignment> RunBufferAssignment(HloModule* module, in RunBufferAssignment()
100 std::unique_ptr<BufferAssignment> RunBufferAssignmentNoBuffersForConstants( in RunBufferAssignmentNoBuffersForConstants()
111 std::unique_ptr<BufferAssignment> RunBufferAssignmentNoBuffersReuseForAdd( in RunBufferAssignmentNoBuffersReuseForAdd()
113 auto reuse_checker = [](const BufferAssignment& assignment, in RunBufferAssignmentNoBuffersReuseForAdd()
129 std::unique_ptr<BufferAssignment> RunColoredBufferAssignment( in RunColoredBufferAssignment()
140 std::unique_ptr<BufferAssignment> RunBufferAssignmentWithInstructionSequence( in RunBufferAssignmentWithInstructionSequence()
243 const BufferAssignment& buffers, HloInstruction* hlo) { in GetAssignedInputAllocation()
254 const BufferAssignment& buffers, HloInstruction* hlo) { in GetAssignedOutputAllocation()
261 const BufferAllocation& GetAllocation(const BufferAssignment& buffers, in GetAllocation()
266 const BufferAllocation& GetTopLevelAllocation(const BufferAssignment& buffers, in GetTopLevelAllocation()
[all …]
|
| D | dump.cc | 185 const BufferAssignment* buffer_assn, in DumpHloModuleImpl()
273 const BufferAssignment& buffer_assn, in DumpHloModuleIfEnabled()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_context.h | 34 const BufferAssignment* buffer_assignment, in IrEmitterContext()
47 const BufferAssignment& buffer_assignment() const { in buffer_assignment()
58 const BufferAssignment* buffer_assignment_;
|
| D | buffer_allocations.h | 52 const BufferAssignment* buffer_assignment, int device_ordinal,
88 const BufferAssignment* buffer_assignment) in BufferAllocations()
108 const BufferAssignment* buffer_assignment_;
|
| D | hlo_to_ir_bindings.h | 39 const BufferAssignment* buffer_assignment, in HloToIrBindings()
103 const BufferAssignment* buffer_assignment_;
|
| D | gpu_executable.h | 58 std::unique_ptr<const BufferAssignment> assignment,
138 const std::unique_ptr<const BufferAssignment> assignment_;
|
| D | buffer_allocations.cc | 41 const BufferAssignment* buffer_assignment, int device_ordinal, in Build()
|
| D | hlo_to_ir_bindings.cc | 195 const BufferAssignment* buffer_assignment) { in BuffersInvariantWithinConsumer()
|
| D | gpu_executable.cc | 55 std::unique_ptr<const BufferAssignment> assignment, in GpuExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | alias_analysis.h | 34 AliasAnalysis(const HloModule& module, const BufferAssignment& assignment, in AliasAnalysis()
61 const BufferAssignment& assignment, const HloInstruction& hlo);
68 const BufferAssignment& assignment_;
|
| D | dynamic_update_slice_util.h | 39 const BufferAssignment& assignment);
44 HloInstruction* fusion, const BufferAssignment& assignment) { in CanEmitFusedDynamicUpdateSliceInPlace()
|
| D | alias_analysis.cc | 125 const BufferAssignment& assignment, const HloInstruction& hlo) { in GetNoaliasMetadataForBuffer()
|
| D | dynamic_update_slice_util.cc | 27 const BufferAssignment& assignment) { in CanUpdateDynamicSliceInPlace()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.h | 51 std::unique_ptr<const BufferAssignment> assignment,
86 const BufferAssignment& buffer_assignment() const { return *assignment_; } in buffer_assignment()
139 const std::unique_ptr<const BufferAssignment> assignment_;
|
| D | buffer_info_util.h | 29 const BufferAssignment& buffer_assignment);
|
| D | buffer_info_util.cc | 24 const BufferAssignment& buffer_assignment) { in CreateBufferInfosFromBufferAssignment()
|
| D | ir_emitter.h | 76 IrEmitter(const HloModule& hlo_module, const BufferAssignment& assignment,
419 const BufferAssignment& assignment_;
|
| D | cpu_executable.cc | 53 std::unique_ptr<const BufferAssignment> assignment, in CpuExecutable()
|
| D | cpu_compiler.cc | 644 std::unique_ptr<BufferAssignment> assignment, in RunBackend()
829 std::unique_ptr<BufferAssignment> assignment, in CompileAheadOfTime()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function.cc | 41 const xla::BufferAssignment& buffer_assignment) { in ComputeResultIndex()
119 const xla::BufferAssignment& buffer_assignment = in Compile()
|