Searched refs:profile_algorithm (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 407 for (const auto& profile_algorithm : algorithms) { in operator ()() local
423 AlgorithmConfig(profile_algorithm), &profile_result); in operator ()()
427 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
429 profile_algorithm.tensor_ops_enabled()); in operator ()()
471 auto profile_algorithm = miopen_algorithm.algorithm(); in operator ()()
476 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in operator ()()
482 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
484 profile_algorithm.tensor_ops_enabled()); in operator ()()
|
| D | conv_ops_3d.cc | 528 for (const auto& profile_algorithm : algorithms) { in launch() local
543 AlgorithmConfig(profile_algorithm), &profile_result); in launch()
548 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in launch()
550 profile_algorithm.tensor_ops_enabled()); in launch()
592 auto profile_algorithm = miopen_algorithm.algorithm(); in launch()
597 AlgorithmConfig(profile_algorithm, in launch()
604 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in launch()
606 profile_algorithm.tensor_ops_enabled()); in launch()
|
| D | conv_grad_ops_3d.cc | 1441 for (const auto& profile_algorithm : algorithms) { in Compute() local
1457 AlgorithmConfig(profile_algorithm), &profile_result); in Compute()
1462 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in Compute()
1464 profile_algorithm.tensor_ops_enabled()); in Compute()
1490 auto profile_algorithm = miopen_algorithm.algorithm(); in Compute()
1495 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in Compute()
1501 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in Compute()
1503 profile_algorithm.tensor_ops_enabled()); in Compute()
1932 for (const auto& profile_algorithm : algorithms) { in Compute() local
1941 AlgorithmConfig(profile_algorithm), &profile_result); in Compute()
[all …]
|
| D | conv_grad_filter_ops.cc | 1001 for (const auto& profile_algorithm : algorithms) { in operator ()() local
1018 AlgorithmConfig(profile_algorithm), &profile_result); in operator ()()
1022 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
1024 profile_algorithm.tensor_ops_enabled()); in operator ()()
1069 auto profile_algorithm = miopen_algorithm.algorithm(); in operator ()()
1074 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in operator ()()
1080 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
1082 profile_algorithm.tensor_ops_enabled()); in operator ()()
|
| D | conv_ops.cc | 1014 for (const auto& profile_algorithm : algorithms) { in operator ()() local
1030 AlgorithmConfig(profile_algorithm), &profile_result); in operator ()()
1034 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
1036 profile_algorithm.tensor_ops_enabled()); in operator ()()
1081 auto profile_algorithm = miopen_algorithm.algorithm(); in operator ()()
1086 AlgorithmConfig(profile_algorithm, miopen_algorithm.scratch_size()), in operator ()()
1091 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
1093 profile_algorithm.tensor_ops_enabled()); in operator ()()
|
| D | conv_ops_fused_impl.h | 394 for (const auto& profile_algorithm : algorithms) {
406 launch(se::dnn::AlgorithmConfig(profile_algorithm), allocator_used,
412 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id());
414 profile_algorithm.tensor_ops_enabled());
|