Searched refs:propagate_nans_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 535 propagate_nans_(false), in PoolingDescriptor()
548 propagate_nans_ = other.propagate_nans_; in CloneFrom()
562 const char* propagate_string = propagate_nans_ ? "Yes" : "No"; in ToString()
578 propagate_nans_ ? "propagate_nans" : "ignore_nans"); in ToShortString()
|
| D | dnn.h | 699 propagate_nans_ = value; in set_propagate_nans()
726 bool propagate_nans() const { return propagate_nans_; } in propagate_nans()
732 bool propagate_nans_; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | maxpooling_op.cc | 377 &propagate_nans_)); in MaxPoolingGradOp()
432 out_backprop, output_shape, propagate_nans_); in Compute()
441 bool propagate_nans_; member in tensorflow::MaxPoolingGradOp
945 &propagate_nans_)); in MaxPoolingWithArgmaxOp()
970 context, params, tensor_in, output, argmax, propagate_nans_, in Compute()
978 bool propagate_nans_; member in tensorflow::MaxPoolingWithArgmaxOp
1187 &propagate_nans_)); in MaxPoolingNoMaskOp()
1214 propagate_nans_); in Compute()
1221 propagate_nans_); in Compute()
1235 output, propagate_nans_); in Compute()
[all …]
|