Searched refs:tpu_executor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_transfer_manager.cc | 95 auto* tpu_executor = static_cast<TpuExecutor*>(executor->implementation()); in TransferLiteralToInfeed() local
98 manager_, tpu_executor->se_executor(), &c_literal, status.c_status); in TransferLiteralToInfeed()
109 auto* tpu_executor = static_cast<TpuExecutor*>(executor->implementation()); in TransferBuffersToInfeed() local
124 manager_, tpu_executor->se_executor(), buffers_array.data(), in TransferBuffersToInfeed()
135 auto* tpu_executor = static_cast<TpuExecutor*>(executor->implementation()); in TransferLiteralFromOutfeed() local
141 manager_, tpu_executor->se_executor(), &c_shape, &c_literal, in TransferLiteralFromOutfeed()
246 auto* tpu_executor = down_cast<TpuExecutor*>(executor->implementation()); in CanShapedBufferBeAccessedNow() local
253 manager_, tpu_executor->se_executor(), &c_device_buffer); in CanShapedBufferBeAccessedNow()
259 auto* tpu_executor = down_cast<TpuExecutor*>(executor->implementation()); in CanBufferBeAccessedNow() local
264 manager_, tpu_executor->se_executor(), &c_device_buffer); in CanBufferBeAccessedNow()
|
| D | BUILD | 100 name = "tpu_executor",
105 "tpu_executor.h",
145 "tpu_executor.h",
177 "tpu_executor.cc",
182 "tpu_executor.h",
245 ":tpu_executor",
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tpu_client.cc | 147 tf_tpu::TpuExecutorInterface* tpu_executor = in GetTpuDevices() local
150 core_id_to_device_ordinal[tpu_executor->GetCoreLocationExternal().Id()] = i; in GetTpuDevices()
|
| D | BUILD | 256 "//tensorflow/stream_executor/tpu:tpu_executor",
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | BUILD | 156 "//tensorflow/stream_executor/tpu:tpu_executor",
180 "//tensorflow/stream_executor/tpu:tpu_executor",
337 "//tensorflow/stream_executor/tpu:tpu_executor",
|