Home
last modified time | relevance | path

Searched refs:buffer_assignment (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dxla_debug_info_manager.cc24 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()
Dxla_debug_info_manager.h58 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
Dxla_debug_info_manager_test.cc32 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()
DBUILD1165 ":buffer_assignment",
1304 name = "buffer_assignment",
1306 "buffer_assignment.cc",
1309 "buffer_assignment.h",
1344 ":buffer_assignment",
4527 ":buffer_assignment",
Dhlo.proto512 BufferAssignmentProto buffer_assignment = 3; field
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_jit_compiled_cpu_function.cc46 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/
Dir_emitter_context.h41 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
Dgpu_compiler.cc556 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 …]
Dhlo_to_ir_bindings.h38 const BufferAssignment* buffer_assignment, in HloToIrBindings() argument
41 : buffer_assignment_(buffer_assignment), in HloToIrBindings()
Dhlo_to_ir_bindings.cc173 const BufferAssignment* buffer_assignment) { in BuffersInvariantWithinConsumer() argument
188 buffer_assignment->HaveDisjointSlices(&hlo, &consumer); in BuffersInvariantWithinConsumer()
Dir_emitter_unnested.cc546 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 …]
DBUILD181 "//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",
Dir_emitter.cc85 &ir_emitter_context->buffer_assignment(), &b_, module_, in IrEmitter()
156 ir_emitter_context_->buffer_assignment().GetUniqueSlice(instr, {}); in EmitConstants()
Dir_emitter_unnested.h344 return ir_emitter_context_->buffer_assignment().GetUniqueSlice(&hlo, index); in MaybeGetAllocationSlice()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dbuffer_info_util.cc25 const BufferAssignment& buffer_assignment) { in CreateBufferInfosFromBufferAssignment() argument
27 for (const BufferAllocation& allocation : buffer_assignment.Allocations()) { in CreateBufferInfosFromBufferAssignment()
Dbuffer_info_util.h29 const BufferAssignment& buffer_assignment);
Dcpu_executable.h82 const BufferAssignment& buffer_assignment() const { return *assignment_; } in buffer_assignment() function
DBUILD124 "//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/
Dinference_context.cc498 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/
DBUILD40 "//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/
Dinference_context.cc545 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/
Druntime.cc600 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/
DBUILD105 "//tensorflow/compiler/xla/service:buffer_assignment",
/external/tensorflow/tensorflow/compiler/mlir/xla/
DBUILD201 "//tensorflow/compiler/xla/service:buffer_assignment",