Searched refs:host_index (Results 1 – 3 of 3) sorted by relevance
283 int64& host_index = *rng_state; in InitializeTypedBuffer() local289 CHECK_LE(host_index, host_buffer->size()); in InitializeTypedBuffer()290 if (host_buffer->size() == host_index) { in InitializeTypedBuffer()291 host_index = 0; in InitializeTypedBuffer()294 std::min<int64>(host_buffer->size() - host_index, elements_left); in InitializeTypedBuffer()296 stream->ThenMemcpy(&mem, host_buffer->data() + host_index, in InitializeTypedBuffer()300 host_index += elements_copied; in InitializeTypedBuffer()
62 host_index = hosts.index(current_host)72 _INDEX_KEY: host_index
357 for host_index in range(num_consumer_hosts):360 consumer_index = host_index * replicas_per_consumer_host + i