Searched refs:profile_algorithm (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 523 auto profile_algorithm = miopen_algorithm.algorithm(); in operator ()() local
528 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in operator ()()
534 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
536 profile_algorithm.tensor_ops_enabled()); in operator ()()
|
| D | conv_ops_3d.cc | 639 auto profile_algorithm = miopen_algorithm.algorithm(); in launch() local
644 AlgorithmConfig(profile_algorithm, in launch()
651 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in launch()
653 profile_algorithm.tensor_ops_enabled()); in launch()
|
| D | conv_grad_ops_3d.cc | 1632 auto profile_algorithm = miopen_algorithm.algorithm(); in Compute() local
1637 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in Compute()
1643 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in Compute()
1645 profile_algorithm.tensor_ops_enabled()); in Compute()
2171 auto profile_algorithm = miopen_algorithm.algorithm(); in Compute() local
2176 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in Compute()
2182 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in Compute()
2184 profile_algorithm.tensor_ops_enabled()); in Compute()
|
| D | conv_grad_filter_ops.cc | 1131 auto profile_algorithm = miopen_algorithm.algorithm(); in operator ()() local
1136 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in operator ()()
1142 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
1144 profile_algorithm.tensor_ops_enabled()); in operator ()()
|
| D | conv_ops.cc | 1250 auto profile_algorithm = miopen_algorithm.algorithm(); in operator ()() local
1255 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in operator ()()
1260 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
1262 profile_algorithm.tensor_ops_enabled()); in operator ()()
|