Home
last modified time | relevance | path

Searched defs:propagate_nans (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_common.cc160 bool propagate_nans) { in Compute()
290 const TensorShape& tensor_in_shape, bool propagate_nans) { in Compute()
Dmaxpooling_op_gpu.cu.cc408 int64* mask, const Eigen::GpuDevice& d, bool propagate_nans, in operator ()()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc735 bool propagate_nans = pooling_descriptor.propagate_nans(); in CudnnPoolingDescriptor() local