Home
last modified time | relevance | path

Searched refs:ThenPoolForward (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_pooling_gpu.cc110 ->ThenPoolForward(pooling_desc, input_desc, input_data, output_desc, in Compute()
115 ->ThenPoolForward(pooling_desc, input_desc, input_data, in Compute()
Dpooling_ops_common.cc398 ->ThenPoolForward(pooling_desc, input_desc, input_data, output_desc, in Compute()
403 ->ThenPoolForward(pooling_desc, input_desc, input_data, in Compute()
/external/tensorflow/tensorflow/stream_executor/
Dstream.h540 Stream &ThenPoolForward(const dnn::PoolingDescriptor &pooling_dimensions,
547 Stream &ThenPoolForward(const dnn::PoolingDescriptor &pooling_dimensions,
554 Stream &ThenPoolForward(const dnn::PoolingDescriptor &pooling_dimensions,
561 Stream &ThenPoolForward(const dnn::PoolingDescriptor &pooling_dimensions,
Dstream.cc819 Stream &Stream::ThenPoolForward( in ThenPoolForward() function in stream_executor::Stream
841 Stream &Stream::ThenPoolForward( in ThenPoolForward() function in stream_executor::Stream
861 Stream &Stream::ThenPoolForward( in ThenPoolForward() function in stream_executor::Stream
882 Stream &Stream::ThenPoolForward( in ThenPoolForward() function in stream_executor::Stream