Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_blas.cc751 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
761 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
772 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
783 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
794 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
804 int ldc, blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
815 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
827 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
839 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
850 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
[all …]
H A Dcuda_dnn.cc1946 dnn::ProfileResult* output_profile_result) { in DoRnnForwardImpl()
2142 port::Status CudnnSupport::DoRnnBackwardImpl( in DoRnnBackwardImpl()
2487 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2530 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2574 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2600 bool CudnnSupport::DoRnnBackward( in DoRnnBackward()
2653 bool CudnnSupport::DoRnnBackward( in DoRnnBackward()
2705 bool CudnnSupport::DoRnnBackward( in DoRnnBackward()
5767 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h1314 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve()
2131 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2153 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2175 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2219 virtual bool DoRnnBackward( in DoRnnBackward()
2248 virtual bool DoRnnBackward( in DoRnnBackward()
2277 virtual bool DoRnnBackward( in DoRnnBackward()
2431 dnn::ProfileResult* output_profile_result) { in DoFusedConvolutionBiasActivation()
2470 dnn::ProfileResult* output_profile_result) { in DoFusedBatchNormActivationInference()
2483 dnn::ProfileResult* output_profile_result) { in DoFusedBatchNormActivationInference()
[all …]
H A Dstream.cc1380 blas::ProfileResult *output_profile_result) { in ThenBlasGemvWithProfiling()
1397 blas::ProfileResult *output_profile_result) { in ThenBlasGemvWithProfiling()
1416 blas::ProfileResult *output_profile_result) { in ThenBlasGemvWithProfiling()
1436 blas::ProfileResult *output_profile_result) { in ThenBlasGemvWithProfiling()
1456 blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithProfiling()
1475 int ldc, blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithProfiling()
1495 blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithProfiling()
1516 blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithProfiling()
1538 blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithProfiling()
2048 dnn::ProfileResult *output_profile_result) { in ThenRnnForward()
[all …]
H A Dstream.h346 dnn::ProfileResult *output_profile_result) { in ConvolveWithAlgorithm()
379 dnn::ProfileResult *output_profile_result) { in FusedConvolveWithAlgorithm()
871 blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithAlgorithm()
886 blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithAlgorithm()
925 blas::ProfileResult *output_profile_result) { in ThenBlasGemmStridedBatchedWithAlgorithm()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_blas.cc516 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
526 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
537 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
548 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
559 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
569 int ldc, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
580 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
592 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
604 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
615 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfilingImpl()
[all …]
H A Drocm_dnn.cc2649 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2692 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2736 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2741 bool MIOpenSupport::DoRnnBackward( in DoRnnBackward()
2794 bool MIOpenSupport::DoRnnBackward( in DoRnnBackward()
2846 bool MIOpenSupport::DoRnnBackward( in DoRnnBackward()
3120 dnn::ProfileResult* output_profile_result) { in DoConvolve()
3809 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve()
4566 dnn::ProfileResult* output_profile_result) { in DoFusedConvolutionBiasActivationImpl()
4655 dnn::ProfileResult* output_profile_result) { in DoFusedConvolutionBiasActivation()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcudnn_rnn_ops.cc801 ProfileResult* output_profile_result) { in DoForward()
884 ProfileResult* output_profile_result) { in DoBackward()