Home
last modified time | relevance | path

Searched refs:ProfileResult (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h117 dnn::ProfileResult* output_profile_result) override;
135 dnn::ProfileResult* output_profile_result) override;
153 dnn::ProfileResult* output_profile_result) override;
178 dnn::ProfileResult* output_profile_result) override;
203 dnn::ProfileResult* output_profile_result) override;
228 dnn::ProfileResult* output_profile_result) override;
243 std::vector<dnn::ProfileResult>* out_algorithms) override;
316 dnn::ProfileResult* output_profile_result) override;
330 dnn::ProfileResult* output_profile_result) override;
344 dnn::ProfileResult* output_profile_result) override;
[all …]
Drocm_blas.h175 blas::ProfileResult *output_profile_result);
183 DeviceMemory<T> *c, int ldc, blas::ProfileResult *output_profile_result);
192 blas::ProfileResult *output_profile_result);
Drocm_dnn.cc504 dnn::ProfileResult GetProfileResultFromConvSolution( in GetProfileResultFromConvSolution()
506 dnn::ProfileResult profile_result; in GetProfileResultFromConvSolution()
514 dnn::ProfileResult GetProfileResultFromConvAlgoPerf( in GetProfileResultFromConvAlgoPerf()
516 dnn::ProfileResult profile_result; in GetProfileResultFromConvAlgoPerf()
2594 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2636 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2679 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2708 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2759 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2811 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
[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.h90 dnn::ProfileResult* output_profile_result) override;
108 dnn::ProfileResult* output_profile_result) override;
126 dnn::ProfileResult* output_profile_result) override;
151 dnn::ProfileResult* output_profile_result) override;
176 dnn::ProfileResult* output_profile_result) override;
201 dnn::ProfileResult* output_profile_result) override;
278 dnn::ProfileResult* output_profile_result) override;
292 dnn::ProfileResult* output_profile_result) override;
306 dnn::ProfileResult* output_profile_result) override;
322 dnn::ProfileResult* output_profile_result) override;
[all …]
Dcuda_blas.h123 blas::ProfileResult *output_profile_result);
131 DeviceMemory<T> *c, int ldc, blas::ProfileResult *output_profile_result);
140 blas::ProfileResult *output_profile_result);
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 …]
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()
2275 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2326 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2378 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
3047 dnn::ProfileResult* output_profile_result) { in DoConvolve()
3190 dnn::ProfileResult* output_profile_result) { in DoFusedConvolveImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h774 class ProfileResult {
1155 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()
1235 dnn::ProfileResult* /*output_profile_result*/) { in DoFusedConvolve() argument
1302 ProfileResult* output_profile_result) = 0;
1314 ProfileResult* output_profile_result) { in DoConvolve()
1339 std::vector<ProfileResult>* out_algorithms);
1417 ProfileResult* output_profile_result) { in DoConvolveBackwardData()
[all …]
Dstream.h71 class ProfileResult; variable
335 dnn::ProfileResult *output_profile_result) { in ConvolveWithAlgorithm()
366 dnn::ProfileResult *output_profile_result);
380 dnn::ProfileResult *output_profile_result);
395 dnn::ProfileResult *output_profile_result);
409 dnn::ProfileResult *output_profile_result);
423 dnn::ProfileResult *output_profile_result);
446 dnn::ProfileResult *output_profile_result) { in ConvolveBackwardDataWithAlgorithm()
478 dnn::ProfileResult *output_profile_result) { in ConvolveBackwardFilterWithAlgorithm()
966 blas::ProfileResult *output_profile_result);
[all …]
Dblas.h163 class ProfileResult {
553 ProfileResult *output_profile_result) = 0;
558 ProfileResult *output_profile_result) = 0;
564 ProfileResult *output_profile_result) = 0;
570 int incy, ProfileResult *output_profile_result) = 0;
1059 ProfileResult *output_profile_result) = 0;
1064 int ldc, ProfileResult *output_profile_result) = 0;
1070 ProfileResult *output_profile_result) = 0;
1077 ProfileResult *output_profile_result) = 0;
1084 ProfileResult *output_profile_result) = 0;
[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()
3242 Stream *, Args..., blas::ProfileResult *), in operator ()()
3243 Args... args, blas::ProfileResult *profile_result) { in operator ()()
3244 ThenBlasImpl<Args..., blas::ProfileResult *> Runner; in operator ()()
3255 blas::ProfileResult *output_profile_result) { in ThenBlasGemvWithProfiling()
3272 blas::ProfileResult *output_profile_result) { in ThenBlasGemvWithProfiling()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_algorithm_picker.cc124 StatusOr<std::vector<se::dnn::ProfileResult>> GetMIOpenAlgorithms( in GetMIOpenAlgorithms()
129 std::vector<se::dnn::ProfileResult> algorithms; in GetMIOpenAlgorithms()
465 se::dnn::ProfileResult profile_result; in PickBestAlgorithmNoCacheCuda()
692 std::vector<se::dnn::ProfileResult> algorithms, in PickBestAlgorithmNoCacheRocm()
718 se::dnn::ProfileResult profile_result; in PickBestAlgorithmNoCacheRocm()
Dgemm_thunk.h85 se::blas::ProfileResult* profile_result = nullptr,
Dgpu_conv_runner.h36 se::dnn::ProfileResult* profile_result = nullptr;
Dgemm_thunk.cc86 se::blas::ProfileResult *output_profile_result) { in DoGemmWithAlgorithm()
177 se::blas::ProfileResult *profile_result, in RunGemm()
Dgemm_algorithm_picker.cc128 se::blas::ProfileResult profile_result; in DoUncachedGemmAutotune()
Dgpu_conv_runner.cc41 using se::dnn::ProfileResult;
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc74 using se::dnn::ProfileResult; in operator ()()
419 ProfileResult profile_result; in operator ()()
443 std::vector<ProfileResult> algorithms; in operator ()()
472 ProfileResult profile_result; in operator ()()
Dconv_ops_3d.cc497 using se::dnn::ProfileResult; in launch()
539 ProfileResult profile_result; in launch()
566 std::vector<ProfileResult> algorithms; in launch()
593 ProfileResult profile_result; in launch()
Dconv_grad_ops_3d.cc1417 using se::dnn::ProfileResult; in Compute()
1453 ProfileResult profile_result; in Compute()
1482 std::vector<ProfileResult> algorithms; in Compute()
1491 ProfileResult profile_result; in Compute()
1914 using se::dnn::ProfileResult; in Compute()
1937 ProfileResult profile_result; in Compute()
1958 std::vector<ProfileResult> algorithms; in Compute()
1968 ProfileResult profile_result; in Compute()
Dcudnn_rnn_ops.cc126 using se::dnn::ProfileResult;
798 ProfileResult* output_profile_result) { in DoForward()
861 ProfileResult* output_profile_result) { in DoBackward()
1697 ProfileResult best_result; in MaybeAutoTune()
1702 ProfileResult final_profile_result; in MaybeAutoTune()
1704 ProfileResult fwd_profile_result; in MaybeAutoTune()
1705 ProfileResult bak_profile_result; in MaybeAutoTune()
Dconv_grad_filter_ops.cc658 using se::dnn::ProfileResult; in operator ()()
1014 ProfileResult profile_result; in operator ()()
1041 std::vector<ProfileResult> algorithms; in operator ()()
1070 ProfileResult profile_result; in operator ()()
Dconv_ops.cc654 using se::dnn::ProfileResult; in operator ()()
1026 ProfileResult profile_result; in operator ()()
1053 std::vector<ProfileResult> algorithms; in operator ()()
1082 ProfileResult profile_result; in operator ()()
Dconv_ops_fused_impl.h403 se::dnn::ProfileResult profile_result;
672 se::dnn::ProfileResult* profile_result) -> Status {

12