Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc237 bool cudnn_deterministic = false; in RequireCudnnDeterminism() local
240 &cudnn_deterministic)); in RequireCudnnDeterminism()
241 return deterministic_ops || cudnn_deterministic; in RequireCudnnDeterminism()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_algorithm_picker.cc368 bool cudnn_deterministic = false; in RequireCudnnDeterminism() local
371 &cudnn_deterministic)); in RequireCudnnDeterminism()
372 return deterministic_ops || cudnn_deterministic; in RequireCudnnDeterminism()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc650 bool cudnn_deterministic = false; in RequireCudnnDeterminism() local
653 &cudnn_deterministic)); in RequireCudnnDeterminism()
654 return deterministic_ops || cudnn_deterministic; in RequireCudnnDeterminism()