Home
last modified time | relevance | path

Searched refs:maybe_owning_memory (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc471 MaybeOwningDeviceMemory* maybe_owning_memory = in ExecuteAsyncOnStreamImpl() local
486 if (output_info.alias_config->must_alias() && maybe_owning_memory && in ExecuteAsyncOnStreamImpl()
487 !maybe_owning_memory->HasOwnership()) { in ExecuteAsyncOnStreamImpl()
493 if (maybe_owning_memory && maybe_owning_memory->HasOwnership()) { in ExecuteAsyncOnStreamImpl()
495 maybe_owning_memory->Release(); in ExecuteAsyncOnStreamImpl()
501 *maybe_owning_memory = argument_buffer; in ExecuteAsyncOnStreamImpl()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc264 MaybeOwningDeviceMemory* maybe_owning_memory = in CreateResultShapedBuffer() local
266 if (alias->must_alias() && !maybe_owning_memory->HasOwnership()) { in CreateResultShapedBuffer()
273 maybe_owning_memory->Release()) { in CreateResultShapedBuffer()
279 *maybe_owning_memory = argument_buffer; in CreateResultShapedBuffer()