Searched refs:maybe_owning_memory (Results 1 – 2 of 2) sorted by relevance
257 MaybeOwningDeviceMemory* maybe_owning_memory = in CreateResultShapedBuffer() local259 if (alias->must_alias() && !maybe_owning_memory->HasOwnership()) { in CreateResultShapedBuffer()266 maybe_owning_memory->Release()) { in CreateResultShapedBuffer()272 *maybe_owning_memory = argument_buffer; in CreateResultShapedBuffer()
747 MaybeOwningDeviceMemory* maybe_owning_memory = in ExecuteAsyncOnStreamImpl() local762 if (output_info.alias_config->must_alias() && maybe_owning_memory && in ExecuteAsyncOnStreamImpl()763 !maybe_owning_memory->HasOwnership()) { in ExecuteAsyncOnStreamImpl()769 if (maybe_owning_memory && maybe_owning_memory->HasOwnership()) { in ExecuteAsyncOnStreamImpl()771 maybe_owning_memory->Release(); in ExecuteAsyncOnStreamImpl()777 *maybe_owning_memory = argument_buffer; in ExecuteAsyncOnStreamImpl()