Searched refs:propagate_nans (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | maxpooling_op_gpu.cu.cc | 33 template <bool propagate_nans, typename dtype>
35 if (propagate_nans) { in IsGreaterThan()
66 template <bool propagate_nans, typename dtype>
92 if (IsGreaterThan<propagate_nans>(bottom_data_n[idx], maxval)) { in MaxPoolForwardNCHW()
143 template <bool propagate_nans, typename dtype>
170 if (IsGreaterThan<propagate_nans>(bottom_data_n[idx], maxval)) { in MaxPoolForwardNHWC()
379 int64* mask, const Eigen::GpuDevice& d, bool propagate_nans, in operator ()() argument
384 if (propagate_nans) { in operator ()()
|
| D | pooling_ops_common_gpu.h | 48 const TensorShape& tensor_out_shape, bool propagate_nans);
65 const TensorShape& tensor_in_shape, bool propagate_nans);
|
| D | maxpooling_op_gpu.h | 42 const Eigen::GpuDevice& d, bool propagate_nans,
|
| D | pooling_ops_common.cc | 220 bool propagate_nans) { in Compute() argument
360 .set_propagate_nans(propagate_nans); in Compute()
447 bool propagate_nans) { in Compute() argument
675 .set_propagate_nans(propagate_nans); in Compute()
|
| D | maxpooling_op.cc | 917 bool propagate_nans, bool include_batch_in_index) { in launch()
1358 const Tensor& input, Tensor* output, bool propagate_nans) { in launch()
1365 propagate_nans, false); in launch()
1377 bool propagate_nans, bool include_batch_in_index) { in launch()
1385 context->eigen_gpu_device(), propagate_nans, include_batch_in_index); in launch()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 796 bool propagate_nans = pooling_descriptor.propagate_nans(); in CudnnPoolingDescriptor() local
805 propagate_nans ? CUDNN_PROPAGATE_NAN : CUDNN_NOT_PROPAGATE_NAN, nd, in CudnnPoolingDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 726 bool propagate_nans() const { return propagate_nans_; } in propagate_nans() function
|