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.cc509 profile_result.set_elapsed_time_in_ms(solution.time); in GetProfileResultFromConvSolution()
535 profile_result.set_elapsed_time_in_ms(algorithm.time); in GetProfileResultFromConvAlgoPerf()
3067 output_profile_result->set_elapsed_time_in_ms( in DoConvolve()
4686 output_profile_result->set_elapsed_time_in_ms( in DoFusedConvolutionBiasActivationImpl()
4781 output_profile_result->set_elapsed_time_in_ms( in DoFusedBatchNormActivationInferenceImpl()
4893 output_profile_result->set_elapsed_time_in_ms( in DoFusedBatchNormActivationForwardImpl()
5011 output_profile_result->set_elapsed_time_in_ms( in DoFusedBatchNormActivationBackwardImpl()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1878 output_profile_result->set_elapsed_time_in_ms( in DoRnnForwardImpl()
2023 output_profile_result->set_elapsed_time_in_ms( in DoRnnBackwardImpl()
3167 output_profile_result->set_elapsed_time_in_ms( in DoConvolve()
3286 output_profile_result->set_elapsed_time_in_ms( in DoFusedConvolveImpl()
Dcuda_blas.cc2019 output_profile_result->set_elapsed_time_in_ms( in DoBlasInternalImpl()
2051 output_profile_result->set_elapsed_time_in_ms( in DoBlasInternalImpl()
2186 output_profile_result->set_elapsed_time_in_ms( in DoBlasInternalImpl()
3734 output_profile_result->set_elapsed_time_in_ms( in DoBlasInternalImpl()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc1744 final_profile_result.set_elapsed_time_in_ms( in MaybeAutoTune()
1756 best_result.set_elapsed_time_in_ms(total_time); in MaybeAutoTune()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h785 void set_elapsed_time_in_ms(float val) { elapsed_time_in_ms_ = val; } in set_elapsed_time_in_ms() function
Dblas.h170 void set_elapsed_time_in_ms(float val) { elapsed_time_in_ms_ = val; } in set_elapsed_time_in_ms() function