Searched refs:infeed_queue_index (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executor.cc | 247 Status TpuExecutor::WaitForInfeedReady(int32 infeed_queue_index) { in WaitForInfeedReady() argument
250 executor_, infeed_queue_index, status.c_status); in WaitForInfeedReady()
270 Status TpuExecutor::EnqueueInfeed(int32 infeed_queue_index, in EnqueueInfeed() argument
274 executor_, infeed_queue_index, bytes.data(), bytes.size(), in EnqueueInfeed()
|
| D | tpu_executor.h | 101 Status EnqueueInfeed(int32 infeed_queue_index, absl::Span<const uint8> bytes);
154 Status WaitForInfeedReady(int32 infeed_queue_index);
|
| D | tpu_executor_c_api.h | 106 int32_t infeed_queue_index, const uint8_t* data,
112 int32_t infeed_queue_index,
|