| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 324 bool with_winograd_nonfused, in GetConvolveAlgorithms() argument
332 return dnn_support->GetConvolveAlgorithms(with_winograd_nonfused, cc_major, in GetConvolveAlgorithms()
346 bool with_winograd_nonfused, in GetConvolveBackwardDataAlgorithms() argument
355 with_winograd_nonfused, cc_major, cc_minor, out_algorithms); in GetConvolveBackwardDataAlgorithms()
359 bool with_winograd_nonfused, in GetConvolveBackwardFilterAlgorithms() argument
368 with_winograd_nonfused, cc_major, cc_minor, out_algorithms); in GetConvolveBackwardFilterAlgorithms()
|
| D | dnn.cc | 30 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveAlgorithms() argument
40 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardDataAlgorithms() argument
46 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardFilterAlgorithms() argument
|
| D | stream_executor_pimpl.h | 385 bool GetConvolveAlgorithms(bool with_winograd_nonfused,
393 bool with_winograd_nonfused,
399 bool with_winograd_nonfused,
|
| D | dnn.h | 1272 bool with_winograd_nonfused, int cc_major, int cc_minor,
1365 bool with_winograd_nonfused, int cc_major, int cc_minor,
1412 bool with_winograd_nonfused, int cc_major, int cc_minor,
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 203 bool with_winograd_nonfused, int cc_major, int cc_minor,
210 bool with_winograd_nonfused, int cc_major, int cc_minor,
214 bool with_winograd_nonfused, int cc_major, int cc_minor,
|
| D | cuda_dnn.cc | 3043 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveAlgorithms() argument
3067 if (CudnnEnvVar<WinogradNonfused>::IsEnabled() && with_winograd_nonfused) { in GetConvolveAlgorithms()
3104 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardDataAlgorithms() argument
3124 if (CudnnEnvVar<WinogradNonfused>::IsEnabled() && with_winograd_nonfused) { in GetConvolveBackwardDataAlgorithms()
3139 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardFilterAlgorithms() argument
3164 if (CudnnEnvVar<WinogradNonfused>::IsEnabled() && with_winograd_nonfused) { in GetConvolveBackwardFilterAlgorithms()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 194 bool with_winograd_nonfused, int cc_major, int cc_minor,
201 bool with_winograd_nonfused, int cc_major, int cc_minor,
205 bool with_winograd_nonfused, int cc_major, int cc_minor,
|
| D | rocm_dnn.cc | 2949 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveAlgorithms() argument
2970 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardDataAlgorithms() argument
2985 bool with_winograd_nonfused, int cc_major, int cc_minor, in GetConvolveBackwardFilterAlgorithms() argument
|