| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment.h | 289 friend class BufferAssignment; variable
355 class BufferAssignment {
497 BufferAssignment(const HloModule* module, in BufferAssignment() function
581 TF_DISALLOW_COPY_AND_ASSIGN(BufferAssignment);
612 static StatusOr<std::unique_ptr<BufferAssignment>> Run(
633 StatusOr<std::unique_ptr<BufferAssignment>> CreateAssignment(
649 BufferAssignment* assignment);
653 BufferAssignment* assignment);
659 BufferAssignment* assignment);
668 BufferAssignment* assignment);
[all …]
|
| D | buffer_assignment.cc | 394 bool BufferAssignment::HasAllocation(const HloValue& value) const { in HasAllocation()
398 bool BufferAssignment::HasAllocation(const HloBuffer& buffer) const { in HasAllocation()
402 const BufferAllocation& BufferAssignment::GetAssignedAllocation( in GetAssignedAllocation()
408 const BufferAllocation& BufferAssignment::GetAssignedAllocation( in GetAssignedAllocation()
413 BufferAllocation* BufferAssignment::GetMutableAssignedAllocation( in GetMutableAssignedAllocation()
418 std::set<BufferAllocation::Slice> BufferAssignment::GetAllSlices( in GetAllSlices()
430 const BufferAllocation& BufferAssignment::GetAllocation( in GetAllocation()
437 const BufferAllocation* BufferAssignment::GetInstructionAllocation( in GetInstructionAllocation()
451 BufferAllocation* BufferAssignment::GetMutableAllocation( in GetMutableAllocation()
456 bool BufferAssignment::HasAllocationAt(const HloInstruction* instruction, in HasAllocationAt()
[all …]
|
| D | dump.h | 32 class BufferAssignment; variable
71 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 | 90 std::unique_ptr<BufferAssignment> RunBufferAssignment(HloModule* module, in RunBufferAssignment()
100 std::unique_ptr<BufferAssignment> RunBufferAssignmentNoBuffersForConstants( in RunBufferAssignmentNoBuffersForConstants()
110 std::unique_ptr<BufferAssignment> RunBufferAssignmentNoBuffersReuseForAdd( in RunBufferAssignmentNoBuffersReuseForAdd()
124 std::unique_ptr<BufferAssignment> RunColoredBufferAssignment( in RunColoredBufferAssignment()
134 std::unique_ptr<BufferAssignment> RunBufferAssignmentWithInstructionSequence( in RunBufferAssignmentWithInstructionSequence()
147 std::unique_ptr<BufferAssignment> RunBufferAssignmentWithPresetAssignments( in RunBufferAssignmentWithPresetAssignments()
250 const BufferAssignment& buffers, HloInstruction* hlo) { in GetAssignedInputAllocation()
261 const BufferAssignment& buffers, HloInstruction* hlo) { in GetAssignedOutputAllocation()
268 const BufferAllocation& GetAllocation(const BufferAssignment& buffers, in GetAllocation()
273 const BufferAllocation& GetTopLevelAllocation(const BufferAssignment& buffers, in GetTopLevelAllocation()
[all …]
|
| D | dump.cc | 220 const BufferAssignment* buffer_assn, in DumpHloModuleImpl()
417 const BufferAssignment& buffer_assn, in DumpHloModuleIfEnabled()
|
| /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 | 56 const BufferAssignment& assignment);
61 const BufferAssignment& assignment);
|
| D | dynamic_update_slice_util.cc | 58 const BufferAssignment& assignment) { in CanUpdateDynamicSliceInPlace()
67 const BufferAssignment& assignment) { in CanEmitFusedDynamicUpdateSliceInPlace()
|
| D | alias_analysis.cc | 122 const BufferAssignment& assignment, const HloInstruction& hlo) { in GetNoaliasMetadataForBuffer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.h | 51 std::unique_ptr<const BufferAssignment> assignment,
82 const BufferAssignment& buffer_assignment() const { return *assignment_; } in buffer_assignment()
132 const std::unique_ptr<const BufferAssignment> assignment_;
|
| D | buffer_info_util.h | 29 const BufferAssignment& buffer_assignment);
|
| D | buffer_info_util.cc | 25 const BufferAssignment& buffer_assignment) { in CreateBufferInfosFromBufferAssignment()
|
| D | cpu_compiler.h | 138 std::tuple<std::unique_ptr<HloModule>, std::unique_ptr<BufferAssignment>>>
|
| D | cpu_compiler.cc | 588 std::tuple<std::unique_ptr<HloModule>, std::unique_ptr<BufferAssignment>>>
608 std::unique_ptr<BufferAssignment> assignment, in RunHloPassesAndBufferAssignement()
724 std::unique_ptr<BufferAssignment> assignment, in RunBackend()
920 std::unique_ptr<BufferAssignment> assignment, in CompileAheadOfTime()
|
| D | ir_emitter.h | 85 const BufferAssignment& assignment, llvm::Module* llvm_module,
432 const BufferAssignment& assignment_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_context.h | 41 const HloModule* hlo_module, const BufferAssignment* buffer_assignment, in IrEmitterContext()
60 const BufferAssignment& buffer_assignment() const { in buffer_assignment()
89 const BufferAssignment* buffer_assignment_;
|
| D | hlo_to_ir_bindings.h | 38 const BufferAssignment* buffer_assignment, in HloToIrBindings()
101 const BufferAssignment* buffer_assignment_;
|
| D | gpu_compiler.cc | 512 std::tuple<std::unique_ptr<HloModule>, std::unique_ptr<BufferAssignment>>>
533 std::unique_ptr<BufferAssignment> assignment, in RunHloPassesAndBufferAssignement()
556 std::unique_ptr<BufferAssignment>* buffer_assignment, in CompileModuleToLlvmIrImpl()
900 std::unique_ptr<BufferAssignment> buffer_assignment; in RunBackend()
990 std::unique_ptr<BufferAssignment> buffer_assignment; in CompileModuleToLlvmIr()
|
| D | gpu_executable.h | 223 GetOutputInfo(const HloModule& hlo_module, const BufferAssignment& assignment);
|
| D | gpu_compiler.h | 61 std::tuple<std::unique_ptr<HloModule>, std::unique_ptr<BufferAssignment>>>
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | driver.cc | 199 struct BufferAssignment { struct
213 std::string BufferAssignmentToString(const BufferAssignment& assignment) { in BufferAssignmentToString()
232 explicit BufferTable(BufferAssignment assignment) : assignment_(assignment) { in BufferTable()
253 BufferAssignment assignment_;
276 BufferAssignment ParseBufferAssignment(const std::string& fname) { in ParseBufferAssignment()
277 BufferAssignment assignment; in ParseBufferAssignment()
462 BufferAssignment assignment = ParseBufferAssignment(arg); in main()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.h | 44 LhloDialectEmitter(const xla::BufferAssignment& assignment, in LhloDialectEmitter()
252 const xla::BufferAssignment& assignment_;
271 tensorflow::Status HloToLhloModule(const xla::BufferAssignment& assignment,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function.cc | 46 const xla::BufferAssignment& buffer_assignment) { in ComputeResultIndex()
128 const xla::BufferAssignment& buffer_assignment = in Compile()
|