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.h445 bool DoPoolForward(Stream* stream,
453 bool DoPoolForward(Stream* stream,
461 bool DoPoolForward(Stream* stream,
469 bool DoPoolForward(Stream* stream,
Dcuda_dnn.cc4261 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/
Drocm_dnn.h469 bool DoPoolForward(Stream* stream,
477 bool DoPoolForward(Stream* stream,
485 bool DoPoolForward(Stream* stream,
Drocm_dnn.cc4001 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/
Ddnn.h1636 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
Dstream.cc830 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()