Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmaxpooling_op.cc375 &propagate_nans_)); in MaxPoolingGradOp()
423 output_shape, propagate_nans_); in Compute()
438 bool propagate_nans_; member in tensorflow::MaxPoolingGradOp
918 &propagate_nans_)); in MaxPoolingWithArgmaxOp()
938 context, params, tensor_in, output, argmax, propagate_nans_, in Compute()
946 bool propagate_nans_; member in tensorflow::MaxPoolingWithArgmaxOp
1143 &propagate_nans_)); in MaxPoolingNoMaskOp()
1169 out_shape, propagate_nans_); in Compute()
1175 out_shape, propagate_nans_); in Compute()
1185 output, propagate_nans_); in Compute()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc505 propagate_nans_(false), in PoolingDescriptor()
518 propagate_nans_ = other.propagate_nans_; in CloneFrom()
532 const char* propagate_string = propagate_nans_ ? "Yes" : "No"; in ToString()
549 propagate_nans_ ? "propagate_nans" : "ignore_nans"); in ToShortString()
Ddnn.h701 propagate_nans_ = value;
724 bool propagate_nans() const { return propagate_nans_; }
729 bool propagate_nans_;