| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function.cc | 41 const xla::BufferAssignment& buffer_assignment) { in ComputeResultIndex() argument
43 buffer_assignment.GetUniqueTopLevelOutputSlice()); in ComputeResultIndex()
119 const xla::BufferAssignment& buffer_assignment = in Compile() local
120 cpu_executable->buffer_assignment(); in Compile()
124 xla::cpu::CreateBufferInfosFromBufferAssignment(buffer_assignment); in Compile()
128 ComputeResultIndex(buffer_assignment)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_context.h | 34 const BufferAssignment* buffer_assignment, in IrEmitterContext() argument
38 buffer_assignment_(buffer_assignment), in IrEmitterContext()
47 const BufferAssignment& buffer_assignment() const { in buffer_assignment() function
|
| D | buffer_allocations.h | 52 const BufferAssignment* buffer_assignment, int device_ordinal,
88 const BufferAssignment* buffer_assignment) in BufferAllocations() argument
92 buffer_assignment_(buffer_assignment) {} in BufferAllocations()
|
| D | buffer_allocations.cc | 41 const BufferAssignment* buffer_assignment, int device_ordinal, in Build() argument
43 const int64 num_buffers = buffer_assignment->Allocations().size(); in Build()
45 num_buffers, device_ordinal, memory_allocator, buffer_assignment)); in Build()
48 const BufferAllocation& allocation = buffer_assignment->GetAllocation(i); in Build()
|
| D | hlo_to_ir_bindings.h | 39 const BufferAssignment* buffer_assignment, in HloToIrBindings() argument
42 : buffer_assignment_(buffer_assignment), in HloToIrBindings()
|
| D | ir_emitter_unnested.cc | 416 const auto& assn = ir_emitter_context_->buffer_assignment(); in HandleCustomCall()
446 const auto& assn = ir_emitter_context_->buffer_assignment(); in HandleCustomCall()
468 const auto& assn = ir_emitter_context_->buffer_assignment(); in HandleCustomCall()
499 const auto& assn = ir_emitter_context_->buffer_assignment(); in HandleCustomCall()
644 fusion, ir_emitter_context_->buffer_assignment())) { in HandleFusion()
691 const BufferAssignment& buffer_assignment = in HandleCopy() local
692 ir_emitter_context_->buffer_assignment(); in HandleCopy()
695 buffer_assignment.GetUniqueTopLevelSlice(copy->operand(0)).ok()) { in HandleCopy()
745 return ir_emitter_context_->buffer_assignment() in HandleTuple()
1452 tuple_element_buffers.push_back(ir_emitter_context_->buffer_assignment() in HandleAllReduce()
[all …]
|
| D | nvptx_compiler.cc | 667 std::unique_ptr<BufferAssignment> buffer_assignment, in RunBackend()
677 buffer_assignment->ToString()); in RunBackend()
679 DumpHloModuleIfEnabled(*module, *buffer_assignment, "after_optimizations"); in RunBackend()
681 IrEmitterContext ir_emitter_context(module.get(), buffer_assignment.get(), in RunBackend()
794 std::move(module), std::move(buffer_assignment), in RunBackend()
|
| D | hlo_to_ir_bindings.cc | 195 const BufferAssignment* buffer_assignment) { in BuffersInvariantWithinConsumer() argument
210 buffer_assignment->HaveDisjointSlices(&hlo, &consumer); in BuffersInvariantWithinConsumer()
|
| D | BUILD | 129 "//tensorflow/compiler/xla/service:buffer_assignment",
178 "//tensorflow/compiler/xla/service:buffer_assignment",
266 "//tensorflow/compiler/xla/service:buffer_assignment",
315 "//tensorflow/compiler/xla/service:buffer_assignment",
390 "//tensorflow/compiler/xla/service:buffer_assignment",
839 "//tensorflow/compiler/xla/service:buffer_assignment",
|
| D | ir_emitter.h | 140 return ir_emitter_context_->buffer_assignment()
|
| D | ir_emitter.cc | 63 &ir_emitter_context->buffer_assignment(), &b_, module_, in IrEmitter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | buffer_info_util.cc | 24 const BufferAssignment& buffer_assignment) { in CreateBufferInfosFromBufferAssignment() argument
26 for (const BufferAllocation& allocation : buffer_assignment.Allocations()) { in CreateBufferInfosFromBufferAssignment()
|
| D | buffer_info_util.h | 29 const BufferAssignment& buffer_assignment);
|
| D | cpu_executable.h | 86 const BufferAssignment& buffer_assignment() const { return *assignment_; } in buffer_assignment() function
|
| D | BUILD | 66 "//tensorflow/compiler/xla/service:buffer_assignment",
112 "//tensorflow/compiler/xla/service:buffer_assignment",
238 "//tensorflow/compiler/xla/service:buffer_assignment",
284 "//tensorflow/compiler/xla/service:buffer_assignment",
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | BUILD | 37 "//tensorflow/compiler/xla/service:buffer_assignment",
191 "//tensorflow/compiler/xla/service:buffer_assignment",
254 "//tensorflow/compiler/xla/service:buffer_assignment",
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | BUILD | 90 "//tensorflow/compiler/xla/service:buffer_assignment",
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | BUILD | 1113 name = "buffer_assignment",
1115 "buffer_assignment.cc",
1118 "buffer_assignment.h",
1148 ":buffer_assignment",
3419 ":buffer_assignment",
|
| D | hlo.proto | 454 BufferAssignmentProto buffer_assignment = 3; field
|