Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc231 bool RequireCudnnDeterminism() { in RequireCudnnDeterminism() function
262 if (!RequireCudnnDeterminism()) { in BestCudnnConvAlgorithm()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_algorithm_picker.cc362 static bool RequireCudnnDeterminism() { in RequireCudnnDeterminism() function
639 if (!RequireCudnnDeterminism() && in PickBestAlgorithmNoCacheCuda()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc644 bool RequireCudnnDeterminism() { in RequireCudnnDeterminism() function
797 const auto cudnn_max_pooling_mode = RequireCudnnDeterminism() in CudnnPoolingDescriptor()
3388 if (!RequireCudnnDeterminism()) { in GetConvolveBackwardDataAlgorithms()
3431 if (!RequireCudnnDeterminism()) { in GetConvolveBackwardFilterAlgorithms()
3907 auto algo = RequireCudnnDeterminism() ? CUDNN_CTC_LOSS_ALGO_DETERMINISTIC in DoPrepareForCtcLoss()
3918 if (RequireCudnnDeterminism()) { in DoPrepareForCtcLoss()