Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2341 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2383 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2426 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2431 bool MIOpenSupport::DoRnnBackward( in DoRnnBackward()
2506 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2534 bool MIOpenSupport::DoRnnBackward( in DoRnnBackward()
2817 dnn::ProfileResult* output_profile_result) { in DoConvolve()
3157 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve()
3174 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve()
3192 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve()
[all …]
Drocm_blas.cc1583 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1593 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1604 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1615 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1626 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1636 int ldc, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1647 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1659 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1671 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1682 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfilingImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc452 dnn::ProfileResult *output_profile_result) { in ThenFusedConvolveWithAlgorithm()
490 dnn::ProfileResult *output_profile_result) { in ThenFusedConvolveWithAlgorithm()
529 dnn::ProfileResult *output_profile_result) { in ThenFusedConvolveWithAlgorithm()
567 dnn::ProfileResult *output_profile_result) { in ThenFusedConvolveWithAlgorithm()
602 dnn::ProfileResult *output_profile_result) { in ThenConvolveWithAlgorithm()
644 dnn::ProfileResult *output_profile_result) { in ThenConvolveWithAlgorithm()
686 dnn::ProfileResult *output_profile_result) { in ThenConvolveWithAlgorithm()
831 dnn::ProfileResult *output_profile_result) { in ThenConvolveBackwardDataWithAlgorithm()
876 dnn::ProfileResult *output_profile_result) { in ThenConvolveBackwardDataWithAlgorithm()
921 dnn::ProfileResult *output_profile_result) { in ThenConvolveBackwardDataWithAlgorithm()
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1522 dnn::ProfileResult* output_profile_result) { in DoRnnForwardImpl()
1682 dnn::ProfileResult* output_profile_result) { in DoRnnBackwardImpl()
1888 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
1929 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
1971 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
1996 bool CudnnSupport::DoRnnBackward( in DoRnnBackward()
2071 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2099 bool CudnnSupport::DoRnnBackward( in DoRnnBackward()
2691 dnn::ProfileResult* output_profile_result) { in DoConvolve()
2935 dnn::ProfileResult* output_profile_result) { in DoFusedConvolveImpl()
[all …]
Dcuda_blas.cc1704 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1714 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1725 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1736 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1747 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1757 int ldc, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1768 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1780 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1792 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1803 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfilingImpl()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc120 se::blas::ProfileResult* output_profile_result) { in DoGemmWithAlgorithm()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc740 ProfileResult* output_profile_result) { in DoForward()
802 ProfileResult* output_profile_result) { in DoBackward()