Home
last modified time | relevance | path

Searched refs:DoPoolForward (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h426 bool DoPoolForward(Stream* stream,
434 bool DoPoolForward(Stream* stream,
442 bool DoPoolForward(Stream* stream,
450 bool DoPoolForward(Stream* stream,
Dcuda_dnn.cc3728 bool CudnnSupport::DoPoolForward( in DoPoolForward() function in stream_executor::gpu::CudnnSupport
3753 bool CudnnSupport::DoPoolForward( in DoPoolForward() function in stream_executor::gpu::CudnnSupport
3778 bool CudnnSupport::DoPoolForward( in DoPoolForward() function in stream_executor::gpu::CudnnSupport
3803 bool CudnnSupport::DoPoolForward( in DoPoolForward() function in stream_executor::gpu::CudnnSupport
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h417 bool DoPoolForward(Stream* stream,
425 bool DoPoolForward(Stream* stream,
433 bool DoPoolForward(Stream* stream,
Drocm_dnn.cc3475 bool MIOpenSupport::DoPoolForward( in DoPoolForward() function in stream_executor::gpu::MIOpenSupport
3485 bool MIOpenSupport::DoPoolForward( in DoPoolForward() function in stream_executor::gpu::MIOpenSupport
3514 bool MIOpenSupport::DoPoolForward( in DoPoolForward() function in stream_executor::gpu::MIOpenSupport
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1570 virtual bool DoPoolForward(Stream* stream,
1578 virtual bool DoPoolForward(Stream* stream,
1589 virtual bool DoPoolForward(Stream* stream,
1600 virtual bool DoPoolForward(Stream* stream,
Dstream.cc1232 CheckError(dnn->DoPoolForward(this, pooling_dimensions, input_dimensions, in ThenPoolForward()
1257 CheckError(dnn->DoPoolForward(this, pooling_dimensions, input_dimensions, in ThenPoolForward()
1280 CheckError(dnn->DoPoolForward(this, pooling_dimensions, input_dimensions, in ThenPoolForward()
1302 CheckError(dnn->DoPoolForward(this, pooling_dimensions, input_dimensions, in ThenPoolForward()