Searched defs:tpu_executor (Results 1 – 2 of 2) sorted by relevance
95 auto* tpu_executor = static_cast<TpuExecutor*>(executor->implementation()); in TransferLiteralToInfeed() local109 auto* tpu_executor = static_cast<TpuExecutor*>(executor->implementation()); in TransferBuffersToInfeed() local135 auto* tpu_executor = static_cast<TpuExecutor*>(executor->implementation()); in TransferLiteralFromOutfeed() local246 auto* tpu_executor = down_cast<TpuExecutor*>(executor->implementation()); in CanShapedBufferBeAccessedNow() local259 auto* tpu_executor = down_cast<TpuExecutor*>(executor->implementation()); in CanBufferBeAccessedNow() local
147 tf_tpu::TpuExecutorInterface* tpu_executor = in GetTpuDevices() local