| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| gemm_thunk.cc | 135 se::blas::ProfileResult profile_result; local
142 &profile_result));
144 if (profile_result.is_valid() && profile_result.elapsed_time_in_ms() <
146 best_result = profile_result;
|
| cudnn_convolution_algorithm_picker.cc | 220 se::dnn::ProfileResult profile_result; local
228 dnums, AlgorithmConfig(alg), &stream, &profile_result)
231 if (launch_ok && profile_result.is_valid()) {
234 << " succeeded, taking " << profile_result.elapsed_time_in_ms()
239 if (profile_result.elapsed_time_in_ms() <
241 best_result = profile_result;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| conv_grad_ops_3d.cc | 673 ProfileResult profile_result; variable
1040 ProfileResult profile_result; variable
[all...] |
| conv_ops_3d.cc | 406 ProfileResult profile_result; local
412 AlgorithmConfig(profile_algorithm), &profile_result)
415 if (profile_result.is_valid()) {
416 if (profile_result.elapsed_time_in_ms() <
418 best_result = profile_result;
421 profile_result.elapsed_time_in_ms() <
423 best_result_no_scratch = profile_result;
|
| matmul_op.cc | 306 ProfileResult profile_result; local
319 &profile_result)
322 if (profile_result.is_valid()) {
323 if (profile_result.elapsed_time_in_ms() <
325 best_result = profile_result;
336 &c_ptr, n, &profile_result)
339 if (profile_result.is_valid()) {
340 if (profile_result.elapsed_time_in_ms() <
342 best_result = profile_result;
350 a_ptr, b_ptr, &c_ptr, &profile_result);
[all...] |
| conv_ops.cc | 719 ProfileResult profile_result; local
725 AlgorithmConfig(profile_algorithm), &profile_result)
728 if (profile_result.is_valid()) {
729 if (profile_result.elapsed_time_in_ms() <
731 best_result = profile_result;
734 profile_result.elapsed_time_in_ms() <
736 best_result_no_scratch = profile_result;
[all...] |
| conv_grad_filter_ops.cc | 920 ProfileResult profile_result; local
[all...] |
| conv_grad_input_ops.cc | 969 ProfileResult profile_result; local
[all...] |
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| fused_conv2d_bias_activation_op.cc | 554 dnn::ProfileResult profile_result; local
563 &profile_result)
566 if (profile_result.is_valid()) {
567 if (profile_result.elapsed_time_in_ms() <
569 best_result = profile_result;
572 profile_result.elapsed_time_in_ms() <
574 best_result_no_scratch = profile_result;
|
