Searched defs:tpu_executor (Results 1 – 2 of 2) sorted by relevance
96 auto* tpu_executor = static_cast<TpuExecutor*>(executor->implementation()); in TransferLiteralToInfeed() local110 auto* tpu_executor = static_cast<TpuExecutor*>(executor->implementation()); in TransferBuffersToInfeed() local136 auto* tpu_executor = static_cast<TpuExecutor*>(executor->implementation()); in TransferLiteralFromOutfeed() local247 auto* tpu_executor = down_cast<TpuExecutor*>(executor->implementation()); in CanShapedBufferBeAccessedNow() local260 auto* tpu_executor = down_cast<TpuExecutor*>(executor->implementation()); in CanBufferBeAccessedNow() local
205 tf_tpu::TpuExecutorInterface* tpu_executor = in GetTpuDevices() local