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()
|