Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h80 dnn::ProfileResult* output_profile_result) override;
98 dnn::ProfileResult* output_profile_result) override;
116 dnn::ProfileResult* output_profile_result) override;
141 dnn::ProfileResult* output_profile_result) override;
166 dnn::ProfileResult* output_profile_result) override;
191 dnn::ProfileResult* output_profile_result) override;
262 dnn::ProfileResult* output_profile_result) override;
276 dnn::ProfileResult* output_profile_result) override;
290 dnn::ProfileResult* output_profile_result) override;
308 dnn::ProfileResult* output_profile_result) override;
[all …]
Drocm_blas.h124 blas::ProfileResult *output_profile_result);
132 DeviceMemory<T> *c, int ldc, blas::ProfileResult *output_profile_result);
141 blas::ProfileResult *output_profile_result);
Drocm_dnn.cc2341 dnn::ProfileResult* output_profile_result) { in DoRnnForward() argument
2383 dnn::ProfileResult* output_profile_result) { in DoRnnForward() argument
2426 dnn::ProfileResult* output_profile_result) { in DoRnnForward() argument
2455 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2506 dnn::ProfileResult* output_profile_result) { in DoRnnBackward() argument
2558 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2817 dnn::ProfileResult* output_profile_result) { in DoConvolve() argument
2833 const bool is_profiling = output_profile_result != nullptr; in DoConvolve()
2931 output_profile_result->set_algorithm(algotype); in DoConvolve()
2932 output_profile_result->set_elapsed_time_in_ms( in DoConvolve()
[all …]
Drocm_blas.cc1583 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling() argument
1586 output_profile_result); in DoBlasGemvWithProfiling()
1593 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling() argument
1596 output_profile_result); in DoBlasGemvWithProfiling()
1604 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling() argument
1607 output_profile_result); in DoBlasGemvWithProfiling()
1615 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling() argument
1618 output_profile_result); in DoBlasGemvWithProfiling()
1626 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling() argument
1629 output_profile_result); in DoBlasGemmWithProfiling()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1127 dnn::ProfileResult* output_profile_result) {
1144 dnn::ProfileResult* output_profile_result) {
1164 dnn::ProfileResult* output_profile_result) {
1182 dnn::ProfileResult* output_profile_result) {
1247 ProfileResult* output_profile_result) = 0;
1259 ProfileResult* output_profile_result) {
1265 output_profile_result),
1266 !output_profile_result);
1351 ProfileResult* output_profile_result) {
1358 output_profile_result),
[all …]
Dstream.h313 dnn::ProfileResult *output_profile_result);
324 dnn::ProfileResult *output_profile_result);
335 dnn::ProfileResult *output_profile_result);
349 dnn::ProfileResult *output_profile_result);
363 dnn::ProfileResult *output_profile_result);
378 dnn::ProfileResult *output_profile_result);
392 dnn::ProfileResult *output_profile_result);
414 dnn::ProfileResult *output_profile_result);
426 dnn::ProfileResult *output_profile_result);
438 dnn::ProfileResult *output_profile_result);
[all …]
Dstream.cc452 dnn::ProfileResult *output_profile_result) { in ThenFusedConvolveWithAlgorithm() argument
467 algorithm_config, output_profile_result); in ThenFusedConvolveWithAlgorithm()
468 if (!status && !output_profile_result) { in ThenFusedConvolveWithAlgorithm()
490 dnn::ProfileResult *output_profile_result) { in ThenFusedConvolveWithAlgorithm() argument
505 algorithm_config, output_profile_result); in ThenFusedConvolveWithAlgorithm()
506 if (!status && !output_profile_result) { in ThenFusedConvolveWithAlgorithm()
529 dnn::ProfileResult *output_profile_result) { in ThenFusedConvolveWithAlgorithm() argument
544 algorithm_config, output_profile_result); in ThenFusedConvolveWithAlgorithm()
545 if (!status && !output_profile_result) { in ThenFusedConvolveWithAlgorithm()
567 dnn::ProfileResult *output_profile_result) { in ThenFusedConvolveWithAlgorithm() argument
[all …]
Dblas.h484 ProfileResult *output_profile_result) = 0;
489 ProfileResult *output_profile_result) = 0;
495 ProfileResult *output_profile_result) = 0;
501 int incy, ProfileResult *output_profile_result) = 0;
990 ProfileResult *output_profile_result) = 0;
995 int ldc, ProfileResult *output_profile_result) = 0;
1001 ProfileResult *output_profile_result) = 0;
1008 ProfileResult *output_profile_result) = 0;
1015 ProfileResult *output_profile_result) = 0;
1041 ProfileResult *output_profile_result) = 0;
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h89 dnn::ProfileResult* output_profile_result) override;
107 dnn::ProfileResult* output_profile_result) override;
125 dnn::ProfileResult* output_profile_result) override;
150 dnn::ProfileResult* output_profile_result) override;
175 dnn::ProfileResult* output_profile_result) override;
200 dnn::ProfileResult* output_profile_result) override;
271 dnn::ProfileResult* output_profile_result) override;
285 dnn::ProfileResult* output_profile_result) override;
299 dnn::ProfileResult* output_profile_result) override;
317 dnn::ProfileResult* output_profile_result) override;
[all …]
Dcuda_dnn.cc1522 dnn::ProfileResult* output_profile_result) { in DoRnnForwardImpl() argument
1555 const bool is_profiling = output_profile_result != nullptr; in DoRnnForwardImpl()
1650 output_profile_result->set_algorithm(algo_desc); in DoRnnForwardImpl()
1651 output_profile_result->set_elapsed_time_in_ms( in DoRnnForwardImpl()
1682 dnn::ProfileResult* output_profile_result) { in DoRnnBackwardImpl() argument
1698 const bool is_profiling = output_profile_result != nullptr; in DoRnnBackwardImpl()
1807 output_profile_result->set_algorithm(algo_desc); in DoRnnBackwardImpl()
1808 output_profile_result->set_elapsed_time_in_ms( in DoRnnBackwardImpl()
1888 dnn::ProfileResult* output_profile_result) { in DoRnnForward() argument
1909 reserve_space_allocator, workspace_allocator, output_profile_result), in DoRnnForward()
[all …]
Dcuda_blas.h126 blas::ProfileResult *output_profile_result);
134 DeviceMemory<T> *c, int ldc, blas::ProfileResult *output_profile_result);
143 blas::ProfileResult *output_profile_result);
Dcuda_blas.cc1704 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling() argument
1707 output_profile_result); in DoBlasGemvWithProfiling()
1714 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling() argument
1717 output_profile_result); in DoBlasGemvWithProfiling()
1725 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling() argument
1728 output_profile_result); in DoBlasGemvWithProfiling()
1736 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling() argument
1739 output_profile_result); in DoBlasGemvWithProfiling()
1747 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling() argument
1750 output_profile_result); in DoBlasGemmWithProfiling()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc120 se::blas::ProfileResult* output_profile_result) { in DoGemmWithAlgorithm() argument
146 algorithm, output_profile_result) in DoGemmWithAlgorithm()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc740 ProfileResult* output_profile_result) { in DoForward() argument
777 workspace_allocator, output_profile_result) in DoForward()
802 ProfileResult* output_profile_result) { in DoBackward() argument
858 workspace_allocator, output_profile_result) in DoBackward()