Searched refs:profile_algorithm (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_3d.cc | 451 for (auto profile_algorithm : algorithms) { in launch() local
461 AlgorithmConfig(profile_algorithm), &profile_result) in launch()
467 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in launch()
469 profile_algorithm.tensor_ops_enabled()); in launch()
|
| D | conv_ops_fused_impl.h | 525 for (auto profile_algorithm : algorithms) {
530 launch(se::dnn::AlgorithmConfig(profile_algorithm), &scratch_allocator,
536 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id());
538 profile_algorithm.tensor_ops_enabled());
|
| D | conv_ops.cc | 861 for (auto profile_algorithm : algorithms) { in operator ()() local
871 AlgorithmConfig(profile_algorithm), &profile_result) in operator ()()
877 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
879 profile_algorithm.tensor_ops_enabled()); in operator ()()
|
| D | conv_grad_filter_ops.cc | 847 for (auto profile_algorithm : algorithms) { in operator ()() local
858 &scratch_allocator, AlgorithmConfig(profile_algorithm), in operator ()()
865 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
867 profile_algorithm.tensor_ops_enabled()); in operator ()()
|
| D | conv_grad_input_ops.cc | 959 for (auto profile_algorithm : algorithms) { in operator ()() local
970 AlgorithmConfig(profile_algorithm), &profile_result) in operator ()()
976 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
978 profile_algorithm.tensor_ops_enabled()); in operator ()()
|
| D | matmul_op.cc | 302 for (auto profile_algorithm : (*algorithms)) { in launch()
313 &c_ptr, n, computation_type, profile_algorithm, in launch()
|
| D | conv_grad_ops_3d.cc | 1367 for (auto profile_algorithm : algorithms) { in Compute() local
1378 AlgorithmConfig(profile_algorithm), &profile_result) in Compute()
1773 for (auto profile_algorithm : algorithms) { in Compute() local
1784 &scratch_allocator, AlgorithmConfig(profile_algorithm), in Compute()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 715 for (auto profile_algorithm : algorithms) { in launch() local
727 dnn::AlgorithmConfig(profile_algorithm), &profile_result) in launch()
733 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in launch()
735 profile_algorithm.tensor_ops_enabled()); in launch()
|