Home
last modified time | relevance | path

Searched refs:MaxPoolForwardWithOptionalArgmax (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmaxpooling_op_gpu.h36 struct MaxPoolForwardWithOptionalArgmax { struct
Dmaxpooling_op_gpu.cu.cc403 bool MaxPoolForwardWithOptionalArgmax<T>::operator()( in operator ()()
519 template struct MaxPoolForwardWithOptionalArgmax<T>; \
Dmaxpooling_op.cc1303 bool status = functor::MaxPoolForwardWithOptionalArgmax<T>()( in launch()
1322 bool status = functor::MaxPoolForwardWithOptionalArgmax<T>()( in launch()