Searched refs:tuple_element_buffer_addresses (Results 1 – 2 of 2) sorted by relevance
29 auto tuple_element_buffer_addresses = absl::make_unique<void*[]>(size); in ExecuteOnStream() local31 tuple_element_buffer_addresses[i] = in ExecuteOnStream()41 tuple_element_buffer_addresses.get(), host_size) in ExecuteOnStream()45 tuple_element_buffer_addresses.get(), dest_buffer_address.opaque(), in ExecuteOnStream()49 auto* buffers_raw = tuple_element_buffer_addresses.release(); in ExecuteOnStream()
74 std::vector<void*> tuple_element_buffer_addresses(tuple_slice.size() / in ExecuteOnStream() local76 stream->ThenMemcpy(tuple_element_buffer_addresses.data(), in ExecuteOnStream()82 se::DeviceMemoryBase(tuple_element_buffer_addresses[index.back()], in ExecuteOnStream()