Searched refs:DoPoolForward (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 426 bool DoPoolForward(Stream* stream, 434 bool DoPoolForward(Stream* stream, 442 bool DoPoolForward(Stream* stream, 450 bool DoPoolForward(Stream* stream,
|
D | cuda_dnn.cc | 3728 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/ |
D | rocm_dnn.h | 417 bool DoPoolForward(Stream* stream, 425 bool DoPoolForward(Stream* stream, 433 bool DoPoolForward(Stream* stream,
|
D | rocm_dnn.cc | 3475 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/ |
D | dnn.h | 1570 virtual bool DoPoolForward(Stream* stream, 1578 virtual bool DoPoolForward(Stream* stream, 1589 virtual bool DoPoolForward(Stream* stream, 1600 virtual bool DoPoolForward(Stream* stream,
|
D | stream.cc | 1232 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()
|