Home
last modified time | relevance | path

Searched defs:output_profile_result (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream.h335 dnn::ProfileResult *output_profile_result) { in ConvolveWithAlgorithm()
446 dnn::ProfileResult *output_profile_result) { in ConvolveBackwardDataWithAlgorithm()
478 dnn::ProfileResult *output_profile_result) { in ConvolveBackwardFilterWithAlgorithm()
Ddnn.h1155 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve()
1173 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve()
1194 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve()
1213 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve()
1314 ProfileResult* output_profile_result) { in DoConvolve()
1417 ProfileResult* output_profile_result) { in DoConvolveBackwardData()
1464 ProfileResult* output_profile_result) { in DoConvolveBackwardFilter()
2202 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2223 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2244 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
[all …]
Dstream.cc458 dnn::ProfileResult *output_profile_result) { in FusedConvolveWithAlgorithm()
489 dnn::ProfileResult *output_profile_result) { in FusedConvolveWithAlgorithm()
521 dnn::ProfileResult *output_profile_result) { in FusedConvolveWithAlgorithm()
552 dnn::ProfileResult *output_profile_result) { in FusedConvolveWithAlgorithm()
583 dnn::ProfileResult *output_profile_result) { in FusedConvolveWithAlgorithm()
3255 blas::ProfileResult *output_profile_result) { in ThenBlasGemvWithProfiling()
3272 blas::ProfileResult *output_profile_result) { in ThenBlasGemvWithProfiling()
3291 blas::ProfileResult *output_profile_result) { in ThenBlasGemvWithProfiling()
3311 blas::ProfileResult *output_profile_result) { in ThenBlasGemvWithProfiling()
3331 blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithProfiling()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2594 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2636 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2679 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2684 bool MIOpenSupport::DoRnnBackward( in DoRnnBackward()
2759 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2787 bool MIOpenSupport::DoRnnBackward( in DoRnnBackward()
2947 dnn::ProfileResult* output_profile_result) { in DoConvolve()
3687 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve()
3703 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve()
3720 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve()
[all …]
Drocm_blas.cc1718 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1728 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1739 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1750 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1761 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1771 int ldc, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1782 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1794 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1806 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1817 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfilingImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1762 dnn::ProfileResult* output_profile_result) { in DoRnnForwardImpl()
1909 dnn::ProfileResult* output_profile_result) { in DoRnnBackwardImpl()
2143 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2184 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2226 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2251 bool CudnnSupport::DoRnnBackward( in DoRnnBackward()
2326 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2354 bool CudnnSupport::DoRnnBackward( in DoRnnBackward()
3047 dnn::ProfileResult* output_profile_result) { in DoConvolve()
3190 dnn::ProfileResult* output_profile_result) { in DoFusedConvolveImpl()
[all …]
Dcuda_blas.cc1899 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
1909 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
1920 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
1931 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
1942 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
1952 int ldc, blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
1963 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
1975 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
1987 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
1998 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc86 se::blas::ProfileResult *output_profile_result) { in DoGemmWithAlgorithm()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc798 ProfileResult* output_profile_result) { in DoForward()
861 ProfileResult* output_profile_result) { in DoBackward()