Searched refs:miopen_algorithm (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 522 for (auto miopen_algorithm : algorithms) { in operator ()() local
523 auto profile_algorithm = miopen_algorithm.algorithm(); in operator ()()
528 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in operator ()()
|
| D | conv_ops_3d.cc | 638 for (auto miopen_algorithm : algorithms) { in launch() local
639 auto profile_algorithm = miopen_algorithm.algorithm(); in launch()
645 miopen_algorithm.scratch_size()), in launch()
|
| D | conv_grad_ops_3d.cc | 1631 for (auto miopen_algorithm : algorithms) { in Compute() local
1632 auto profile_algorithm = miopen_algorithm.algorithm(); in Compute()
1637 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in Compute()
2170 for (auto miopen_algorithm : algorithms) { in Compute() local
2171 auto profile_algorithm = miopen_algorithm.algorithm(); in Compute()
2176 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in Compute()
|
| D | conv_grad_filter_ops.cc | 1130 for (auto miopen_algorithm : algorithms) { in operator ()() local
1131 auto profile_algorithm = miopen_algorithm.algorithm(); in operator ()()
1136 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in operator ()()
|
| D | conv_ops.cc | 1249 for (auto miopen_algorithm : algorithms) { in operator ()() local
1250 auto profile_algorithm = miopen_algorithm.algorithm(); in operator ()()
1255 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in operator ()()
|