Searched defs:tpu_executor (Results 1 – 2 of 2) sorted by relevance
97 auto* tpu_executor = static_cast<TpuExecutor*>(executor->implementation()); in TransferLiteralToInfeed() local111 auto* tpu_executor = static_cast<TpuExecutor*>(executor->implementation()); in TransferBuffersToInfeed() local137 auto* tpu_executor = static_cast<TpuExecutor*>(executor->implementation()); in TransferLiteralFromOutfeed() local248 auto* tpu_executor = down_cast<TpuExecutor*>(executor->implementation()); in CanShapedBufferBeAccessedNow() local262 auto* tpu_executor = down_cast<TpuExecutor*>(executor->implementation()); in CanBufferBeAccessedNow() local
222 tf_tpu::TpuExecutorInterface* tpu_executor = in GetTpuDevices() local