Home
last modified time | relevance | path

Searched refs:output_profile_result (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_blas.cc751 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl() argument
754 output_profile_result); in DoBlasInternalImpl()
761 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl() argument
764 output_profile_result); in DoBlasInternalImpl()
772 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl() argument
775 output_profile_result); in DoBlasInternalImpl()
783 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl() argument
786 output_profile_result); in DoBlasInternalImpl()
794 blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl() argument
797 output_profile_result); in DoBlasInternalImpl()
[all …]
Dcuda_dnn.h98 dnn::ProfileResult* output_profile_result) override;
117 dnn::ProfileResult* output_profile_result) override;
136 dnn::ProfileResult* output_profile_result) override;
162 dnn::ProfileResult* output_profile_result) override;
188 dnn::ProfileResult* output_profile_result) override;
214 dnn::ProfileResult* output_profile_result) override;
345 dnn::ProfileResult* output_profile_result) override;
361 dnn::ProfileResult* output_profile_result) override;
584 dnn::ProfileResult* output_profile_result);
611 dnn::ProfileResult* output_profile_result);
Dcuda_blas.h126 blas::ProfileResult *output_profile_result);
135 blas::ProfileResult *output_profile_result);
Dcuda_dnn.cc1946 dnn::ProfileResult* output_profile_result) { in DoRnnForwardImpl() argument
1990 const bool is_profiling = output_profile_result != nullptr; in DoRnnForwardImpl()
2023 output_profile_result->set_algorithm(algo_desc); in DoRnnForwardImpl()
2024 output_profile_result->set_elapsed_time_in_ms( in DoRnnForwardImpl()
2051 const bool is_profiling = output_profile_result != nullptr; in DoRnnForwardImpl()
2133 output_profile_result->set_algorithm(algo_desc); in DoRnnForwardImpl()
2134 output_profile_result->set_elapsed_time_in_ms( in DoRnnForwardImpl()
2166 dnn::ProfileResult* output_profile_result) { in DoRnnBackwardImpl()
2197 const bool is_profiling = output_profile_result != nullptr; in DoRnnBackwardImpl()
2256 output_profile_result->set_algorithm(algo_desc); in DoRnnBackwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h118 dnn::ProfileResult* output_profile_result) override;
137 dnn::ProfileResult* output_profile_result) override;
156 dnn::ProfileResult* output_profile_result) override;
182 dnn::ProfileResult* output_profile_result) override;
208 dnn::ProfileResult* output_profile_result) override;
234 dnn::ProfileResult* output_profile_result) override;
349 dnn::ProfileResult* output_profile_result) override;
365 dnn::ProfileResult* output_profile_result) override;
532 dnn::ProfileResult* output_profile_result) override;
543 dnn::ProfileResult* output_profile_result) override;
[all …]
Drocm_blas.cc516 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling() argument
519 output_profile_result); in DoBlasGemvWithProfiling()
526 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling() argument
529 output_profile_result); in DoBlasGemvWithProfiling()
537 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling() argument
540 output_profile_result); in DoBlasGemvWithProfiling()
548 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling() argument
551 output_profile_result); in DoBlasGemvWithProfiling()
559 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling() argument
562 output_profile_result); in DoBlasGemmWithProfiling()
[all …]
Drocm_dnn.cc2649 dnn::ProfileResult* output_profile_result) { in DoRnnForward() argument
2692 dnn::ProfileResult* output_profile_result) { in DoRnnForward() argument
2736 dnn::ProfileResult* output_profile_result) { in DoRnnForward() argument
2766 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2818 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2871 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
3120 dnn::ProfileResult* output_profile_result) { in DoConvolve() argument
3127 return (*runner)(stream, output_profile_result, scratch_memory, input_data, in DoConvolve()
3809 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve() argument
4566 dnn::ProfileResult* output_profile_result) { in DoFusedConvolutionBiasActivationImpl() argument
[all …]
Drocm_blas.h183 blas::ProfileResult *output_profile_result);
192 blas::ProfileResult *output_profile_result);
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dblas.h298 ProfileResult *output_profile_result) = 0;
303 ProfileResult *output_profile_result) = 0;
309 ProfileResult *output_profile_result) = 0;
315 int incy, ProfileResult *output_profile_result) = 0;
361 ProfileResult *output_profile_result) = 0;
367 ProfileResult *output_profile_result) = 0;
373 ProfileResult *output_profile_result) = 0;
380 ProfileResult *output_profile_result) = 0;
387 ProfileResult *output_profile_result) = 0;
411 ProfileResult *output_profile_result) = 0;
[all …]
Dstream.h346 dnn::ProfileResult *output_profile_result) { in ConvolveWithAlgorithm() argument
360 scratch_memory, output_profile_result); in ConvolveWithAlgorithm()
379 dnn::ProfileResult *output_profile_result) { in FusedConvolveWithAlgorithm() argument
388 scratch_allocator, algorithm_config, output_profile_result); in FusedConvolveWithAlgorithm()
723 blas::ProfileResult *output_profile_result);
729 blas::ProfileResult *output_profile_result);
735 blas::ProfileResult *output_profile_result);
741 int incy, blas::ProfileResult *output_profile_result);
833 blas::ProfileResult *output_profile_result);
840 blas::ProfileResult *output_profile_result);
[all …]
Dstream.cc1380 blas::ProfileResult *output_profile_result) { in ThenBlasGemvWithProfiling() argument
1390 alpha, a, lda, x, incx, beta, y, incy, output_profile_result); in ThenBlasGemvWithProfiling()
1397 blas::ProfileResult *output_profile_result) { in ThenBlasGemvWithProfiling() argument
1408 alpha, a, lda, x, incx, beta, y, incy, output_profile_result); in ThenBlasGemvWithProfiling()
1416 blas::ProfileResult *output_profile_result) { in ThenBlasGemvWithProfiling() argument
1428 alpha, a, lda, x, incx, beta, y, incy, output_profile_result); in ThenBlasGemvWithProfiling()
1436 blas::ProfileResult *output_profile_result) { in ThenBlasGemvWithProfiling() argument
1448 alpha, a, lda, x, incx, beta, y, incy, output_profile_result); in ThenBlasGemvWithProfiling()
1456 blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithProfiling() argument
1468 output_profile_result); in ThenBlasGemmWithProfiling()
[all …]
Ddnn.h1314 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve() argument
1381 ProfileResult* output_profile_result) = 0;
2131 dnn::ProfileResult* output_profile_result) { in DoRnnForward() argument
2153 dnn::ProfileResult* output_profile_result) { in DoRnnForward() argument
2175 dnn::ProfileResult* output_profile_result) { in DoRnnForward() argument
2244 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2273 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2302 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2431 dnn::ProfileResult* output_profile_result) { in DoFusedConvolutionBiasActivation() argument
2470 dnn::ProfileResult* output_profile_result) { in DoFusedBatchNormActivationInference() argument
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc801 ProfileResult* output_profile_result) { in DoForward() argument
859 output_profile_result) in DoForward()
884 ProfileResult* output_profile_result) { in DoBackward() argument
960 &reserve_space_uint8, workspace_allocator, output_profile_result) in DoBackward()