| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 265 bool with_winograd_nonfused, in GetConvolveAlgorithms() argument
273 return dnn_support->GetConvolveAlgorithms(with_winograd_nonfused, cc_major, in GetConvolveAlgorithms()
306 bool with_winograd_nonfused, in GetConvolveBackwardDataAlgorithms() argument
315 with_winograd_nonfused, cc_major, cc_minor, out_algorithms); in GetConvolveBackwardDataAlgorithms()
319 bool with_winograd_nonfused, in GetConvolveBackwardFilterAlgorithms() argument
328 with_winograd_nonfused, cc_major, cc_minor, out_algorithms); in GetConvolveBackwardFilterAlgorithms()
|
| D | stream_executor_pimpl.h | 365 bool GetConvolveAlgorithms(bool with_winograd_nonfused,
386 bool with_winograd_nonfused,
392 bool with_winograd_nonfused,
|
| D | dnn.cc | 45 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveAlgorithms() argument
69 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardDataAlgorithms() argument
75 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardFilterAlgorithms() argument
|
| D | dnn.h | 1327 bool with_winograd_nonfused, int cc_major, int cc_minor,
1431 bool with_winograd_nonfused, int cc_major, int cc_minor,
1478 bool with_winograd_nonfused, int cc_major, int cc_minor,
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 204 bool with_winograd_nonfused, int cc_major, int cc_minor,
211 bool with_winograd_nonfused, int cc_major, int cc_minor,
215 bool with_winograd_nonfused, int cc_major, int cc_minor,
|
| D | cuda_dnn.cc | 3295 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveAlgorithms() argument
3323 if (CudnnEnvVar<WinogradNonfused>::IsEnabled() && with_winograd_nonfused) { in GetConvolveAlgorithms()
3365 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardDataAlgorithms() argument
3385 if (CudnnEnvVar<WinogradNonfused>::IsEnabled() && with_winograd_nonfused) { in GetConvolveBackwardDataAlgorithms()
3404 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardFilterAlgorithms() argument
3428 if (CudnnEnvVar<WinogradNonfused>::IsEnabled() && with_winograd_nonfused) { in GetConvolveBackwardFilterAlgorithms()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 231 bool with_winograd_nonfused, int cc_major, int cc_minor,
249 bool with_winograd_nonfused, int cc_major, int cc_minor,
253 bool with_winograd_nonfused, int cc_major, int cc_minor,
|
| D | rocm_dnn.cc | 3084 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveAlgorithms() argument
3495 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardDataAlgorithms() argument
3510 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardFilterAlgorithms() argument
|