Searched refs:se_client (Results 1 – 1 of 1) sorted by relevance
372 auto* se_client = tensorflow::down_cast<PjRtStreamExecutorClient*>(client); in AllocateDestinationBuffer() local374 se_client->client()->backend().transfer_manager(); in AllocateDestinationBuffer()377 on_host_shape, se_client->allocator(), in AllocateDestinationBuffer()