Searched refs:outfeed_queue_index (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executor.cc | 254 Status TpuExecutor::WaitForOutfeedReady(int32 outfeed_queue_index) { in WaitForOutfeedReady() argument
257 executor_, outfeed_queue_index, status.c_status); in WaitForOutfeedReady()
261 void TpuExecutor::DequeueOutfeed(int32 outfeed_queue_index, in DequeueOutfeed() argument
265 executor_, outfeed_queue_index, bytes.data(), bytes.size(), in DequeueOutfeed()
|
| D | tpu_executor.h | 98 void DequeueOutfeed(int32 outfeed_queue_index, absl::Span<uint8> bytes,
156 Status WaitForOutfeedReady(int32 outfeed_queue_index);
|
| D | tpu_executor_c_api.h | 109 int32_t outfeed_queue_index, uint8_t* data,
115 int32_t outfeed_queue_index,
|