Searched refs:se_executor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_transfer_manager.cc | 98 manager_, tpu_executor->se_executor(), &c_literal, status.c_status); in TransferLiteralToInfeed()
124 manager_, tpu_executor->se_executor(), buffers_array.data(), in TransferBuffersToInfeed()
141 manager_, tpu_executor->se_executor(), &c_shape, &c_literal, in TransferLiteralFromOutfeed()
157 ->se_executor()); in ResetDevices()
253 manager_, tpu_executor->se_executor(), &c_device_buffer); in CanShapedBufferBeAccessedNow()
264 manager_, tpu_executor->se_executor(), &c_device_buffer); in CanBufferBeAccessedNow()
|
| D | tpu_executor.h | 220 SE_StreamExecutor* se_executor() { return executor_; } in se_executor() function
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_on_demand_compiler.cc | 235 ->se_executor(), in RunHloPasses()
264 ->se_executor(), in RunBackend()
305 ->se_executor(); in Compile()
|