Home
last modified time | relevance | path

Searched refs:set_elapsed_time_in_ms (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2932 output_profile_result->set_elapsed_time_in_ms( in DoConvolve()
4090 output_profile_result->set_elapsed_time_in_ms( in DoFusedConvolutionBiasActivationImpl()
4185 output_profile_result->set_elapsed_time_in_ms( in DoFusedBatchNormActivationInferenceImpl()
4297 output_profile_result->set_elapsed_time_in_ms( in DoFusedBatchNormActivationForwardImpl()
4415 output_profile_result->set_elapsed_time_in_ms( in DoFusedBatchNormActivationBackwardImpl()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc1557 final_profile_result.set_elapsed_time_in_ms( in MaybeAutoTune()
1569 best_result.set_elapsed_time_in_ms(total_time); in MaybeAutoTune()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1651 output_profile_result->set_elapsed_time_in_ms( in DoRnnForwardImpl()
1808 output_profile_result->set_elapsed_time_in_ms( in DoRnnBackwardImpl()
2913 output_profile_result->set_elapsed_time_in_ms( in DoConvolve()
3030 output_profile_result->set_elapsed_time_in_ms( in DoFusedConvolveImpl()
Dcuda_blas.cc1824 output_profile_result->set_elapsed_time_in_ms( in DoBlasGemvWithProfilingImpl()
1856 output_profile_result->set_elapsed_time_in_ms( in DoBlasGemmWithProfilingImpl()
1975 output_profile_result->set_elapsed_time_in_ms( in DoBlasGemmWithAlgorithmImpl()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h779 void set_elapsed_time_in_ms(float val) { elapsed_time_in_ms_ = val; }
Dblas.h139 void set_elapsed_time_in_ms(float val) { elapsed_time_in_ms_ = val; } in set_elapsed_time_in_ms() function