/external/tensorflow/tensorflow/compiler/xla/service/ |
D | xla_debug_info_manager.cc | 24 std::shared_ptr<const BufferAssignmentProto> buffer_assignment) { in RegisterModule() argument 28 buffer_assignment); in RegisterModule() 32 m.instances.emplace_back(hlo_module, buffer_assignment); in RegisterModule() 42 std::shared_ptr<const BufferAssignmentProto> buffer_assignment) { in UnregisterModule() argument 49 e.buffer_assignment == buffer_assignment; in UnregisterModule() 147 if (m.instances[0].hlo_module && m.instances[0].buffer_assignment) { in StopTracing() 151 *m.instances[0].buffer_assignment; in StopTracing()
|
D | xla_debug_info_manager.h | 58 std::shared_ptr<const BufferAssignmentProto> buffer_assignment); 65 std::shared_ptr<const BufferAssignmentProto> buffer_assignment); 113 : hlo_module(std::move(m)), buffer_assignment(std::move(b)) {} in XlaModuleInstance() 115 std::shared_ptr<const BufferAssignmentProto> buffer_assignment; member
|
D | xla_debug_info_manager_test.cc | 32 std::shared_ptr<BufferAssignmentProto> buffer_assignment; member 42 debug_info.buffer_assignment = nullptr; in RegisterProgram() 44 debug_info.buffer_assignment); in RegisterProgram() 54 external_references_[i].buffer_assignment); in UnregisterProgram()
|
D | BUILD | 1165 ":buffer_assignment", 1304 name = "buffer_assignment", 1306 "buffer_assignment.cc", 1309 "buffer_assignment.h", 1344 ":buffer_assignment", 4527 ":buffer_assignment",
|
D | hlo.proto | 512 BufferAssignmentProto buffer_assignment = 3; field
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_jit_compiled_cpu_function.cc | 46 const xla::BufferAssignment& buffer_assignment) { in ComputeResultIndex() argument 48 buffer_assignment.GetUniqueTopLevelOutputSlice()); in ComputeResultIndex() 128 const xla::BufferAssignment& buffer_assignment = in Compile() local 129 cpu_executable->buffer_assignment(); in Compile() 133 xla::cpu::CreateBufferInfosFromBufferAssignment(buffer_assignment); in Compile() 137 ComputeResultIndex(buffer_assignment)); in Compile()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emitter_context.h | 41 const HloModule* hlo_module, const BufferAssignment* buffer_assignment, in IrEmitterContext() argument 47 buffer_assignment_(buffer_assignment), in IrEmitterContext() 60 const BufferAssignment& buffer_assignment() const { in buffer_assignment() function
|
D | gpu_compiler.cc | 556 std::unique_ptr<BufferAssignment>* buffer_assignment, in CompileModuleToLlvmIrImpl() argument 576 *buffer_assignment, in CompileModuleToLlvmIrImpl() 587 << (*buffer_assignment)->GetStats().ToString(); in CompileModuleToLlvmIrImpl() 588 DumpHloModuleIfEnabled(*hlo_module, **buffer_assignment, in CompileModuleToLlvmIrImpl() 597 hlo_module, buffer_assignment->get(), platform_name, gpu_device_info, in CompileModuleToLlvmIrImpl() 900 std::unique_ptr<BufferAssignment> buffer_assignment; in RunBackend() local 908 &buffer_assignment, &thunk_schedule, &constants)); in RunBackend() 935 GetOutputInfo(*module, *buffer_assignment)); in RunBackend() 937 std::make_unique<BufferAssignmentProto>(buffer_assignment->ToProto()); in RunBackend() 939 buffer_assignment->ReleaseAllocations(); in RunBackend() [all …]
|
D | hlo_to_ir_bindings.h | 38 const BufferAssignment* buffer_assignment, in HloToIrBindings() argument 41 : buffer_assignment_(buffer_assignment), in HloToIrBindings()
|
D | hlo_to_ir_bindings.cc | 173 const BufferAssignment* buffer_assignment) { in BuffersInvariantWithinConsumer() argument 188 buffer_assignment->HaveDisjointSlices(&hlo, &consumer); in BuffersInvariantWithinConsumer()
|
D | ir_emitter_unnested.cc | 546 emitter->ir_emitter_context_->buffer_assignment(), *hlo_computation, in Create() 1748 const auto& buffer_assignment = ir_emitter_context_->buffer_assignment(); in GetMlirEmitterInput() local 1751 buffer_assignment.GetUniqueSlice(hlo, {})); in GetMlirEmitterInput() 3458 GetHloBufferSlices(inst, ir_emitter_context_->buffer_assignment()); in BuildKernelThunk() 3760 const BufferAssignment& buffer_assignment) { in CheckHloBuffersShareAllocation() argument 3762 buffer_assignment.GetUniqueSlice(a, index).ConsumeValueOrDie(); in CheckHloBuffersShareAllocation() 3764 buffer_assignment.GetUniqueSlice(b, index).ConsumeValueOrDie(); in CheckHloBuffersShareAllocation() 3779 const BufferAssignment& buffer_assignment) { in CheckWhileBuffersShareAllocation() argument 3789 xla_while, xla_while->operand(0), index, buffer_assignment)); in CheckWhileBuffersShareAllocation() 3791 xla_while, condition_parameter, index, buffer_assignment)); in CheckWhileBuffersShareAllocation() [all …]
|
D | BUILD | 181 "//tensorflow/compiler/xla/service:buffer_assignment", 224 "//tensorflow/compiler/xla/service:buffer_assignment", 284 "//tensorflow/compiler/xla/service:buffer_assignment", 379 "//tensorflow/compiler/xla/service:buffer_assignment", 458 "//tensorflow/compiler/xla/service:buffer_assignment", 630 "//tensorflow/compiler/xla/service:buffer_assignment", 689 "//tensorflow/compiler/xla/service:buffer_assignment", 1281 "//tensorflow/compiler/xla/service:buffer_assignment",
|
D | ir_emitter.cc | 85 &ir_emitter_context->buffer_assignment(), &b_, module_, in IrEmitter() 156 ir_emitter_context_->buffer_assignment().GetUniqueSlice(instr, {}); in EmitConstants()
|
D | ir_emitter_unnested.h | 344 return ir_emitter_context_->buffer_assignment().GetUniqueSlice(&hlo, index); in MaybeGetAllocationSlice()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | buffer_info_util.cc | 25 const BufferAssignment& buffer_assignment) { in CreateBufferInfosFromBufferAssignment() argument 27 for (const BufferAllocation& allocation : buffer_assignment.Allocations()) { in CreateBufferInfosFromBufferAssignment()
|
D | buffer_info_util.h | 29 const BufferAssignment& buffer_assignment);
|
D | cpu_executable.h | 82 const BufferAssignment& buffer_assignment() const { return *assignment_; } in buffer_assignment() function
|
D | BUILD | 124 "//tensorflow/compiler/xla/service:buffer_assignment", 186 "//tensorflow/compiler/xla/service:buffer_assignment", 335 "//tensorflow/compiler/xla/service:buffer_assignment", 389 "//tensorflow/compiler/xla/service:buffer_assignment",
|
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
D | inference_context.cc | 498 ObjectsAssignment<size_t> buffer_assignment; in AllocateMemoryForBuffers() local 500 buffer_usage_records, MemoryStrategy::GREEDY_BEST, &buffer_assignment)); in AllocateMemoryForBuffers() 504 shared_buffers_.resize(buffer_assignment.object_sizes.size()); in AllocateMemoryForBuffers() 505 for (int i = 0; i < buffer_assignment.object_sizes.size(); ++i) { in AllocateMemoryForBuffers() 507 NSUInteger bufferSize = dataTypeSize * buffer_assignment.object_sizes[i]; in AllocateMemoryForBuffers() 511 error += std::to_string(buffer_assignment.object_ids[i]) + in AllocateMemoryForBuffers() 534 const int buffer_index = buffer_assignment.object_ids[tensor_index]; in AllocateMemoryForBuffers()
|
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
D | BUILD | 40 "//tensorflow/compiler/xla/service:buffer_assignment", 181 "//tensorflow/compiler/xla/service:buffer_assignment", 245 "//tensorflow/compiler/xla/service:buffer_assignment",
|
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
D | inference_context.cc | 545 ObjectsAssignment<size_t> buffer_assignment; in AllocateMemoryForBuffers() local 547 buffer_usage_records, MemoryStrategy::GREEDY_BEST, &buffer_assignment)); in AllocateMemoryForBuffers() 549 shared_buffers_.resize(buffer_assignment.object_sizes.size()); in AllocateMemoryForBuffers() 550 for (int i = 0; i < buffer_assignment.object_sizes.size(); ++i) { in AllocateMemoryForBuffers() 551 RETURN_IF_ERROR(CreateReadWriteBuffer(buffer_assignment.object_sizes[i], in AllocateMemoryForBuffers() 564 const int buffer_index = buffer_assignment.object_ids[tensor_index]; in AllocateMemoryForBuffers()
|
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
D | runtime.cc | 600 ObjectsAssignment<size_t> buffer_assignment; in AssignInternalObjects() local 603 &buffer_assignment)); in AssignInternalObjects() 605 buffer_assignment, usage_records.usage_refs, global_ref_to_object_ptr, in AssignInternalObjects()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
D | BUILD | 105 "//tensorflow/compiler/xla/service:buffer_assignment",
|
/external/tensorflow/tensorflow/compiler/mlir/xla/ |
D | BUILD | 201 "//tensorflow/compiler/xla/service:buffer_assignment",
|