Searched refs:DoPoolForward (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 445 bool DoPoolForward(Stream* stream,
453 bool DoPoolForward(Stream* stream,
461 bool DoPoolForward(Stream* stream,
469 bool DoPoolForward(Stream* stream,
|
| D | cuda_dnn.cc | 4261 bool CudnnSupport::DoPoolForward( in DoPoolForward() function in stream_executor::gpu::CudnnSupport
4286 bool CudnnSupport::DoPoolForward( in DoPoolForward() function in stream_executor::gpu::CudnnSupport
4311 bool CudnnSupport::DoPoolForward( in DoPoolForward() function in stream_executor::gpu::CudnnSupport
4336 bool CudnnSupport::DoPoolForward( in DoPoolForward() function in stream_executor::gpu::CudnnSupport
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 469 bool DoPoolForward(Stream* stream,
477 bool DoPoolForward(Stream* stream,
485 bool DoPoolForward(Stream* stream,
|
| D | rocm_dnn.cc | 4001 bool MIOpenSupport::DoPoolForward( in DoPoolForward() function in stream_executor::gpu::MIOpenSupport
4095 bool MIOpenSupport::DoPoolForward( in DoPoolForward() function in stream_executor::gpu::MIOpenSupport
4161 bool MIOpenSupport::DoPoolForward( in DoPoolForward() function in stream_executor::gpu::MIOpenSupport
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1636 virtual bool DoPoolForward(Stream* stream,
1644 virtual bool DoPoolForward(Stream* stream, in DoPoolForward() function
1655 virtual bool DoPoolForward(Stream* stream, in DoPoolForward() function
1666 virtual bool DoPoolForward(Stream* stream, in DoPoolForward() function
|
| D | stream.cc | 830 CheckError(dnn->DoPoolForward(this, pooling_dimensions, input_dimensions, in ThenPoolForward()
852 CheckError(dnn->DoPoolForward(this, pooling_dimensions, input_dimensions, in ThenPoolForward()
873 CheckError(dnn->DoPoolForward(this, pooling_dimensions, input_dimensions, in ThenPoolForward()
893 CheckError(dnn->DoPoolForward(this, pooling_dimensions, input_dimensions, in ThenPoolForward()
|