| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 117 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 …]
|
| D | rocm_blas.h | 175 blas::ProfileResult *output_profile_result);
183 DeviceMemory<T> *c, int ldc, blas::ProfileResult *output_profile_result);
192 blas::ProfileResult *output_profile_result);
|
| D | rocm_dnn.cc | 504 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 …]
|
| D | rocm_blas.cc | 1718 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/ |
| D | cuda_dnn.h | 90 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 …]
|
| D | cuda_blas.h | 123 blas::ProfileResult *output_profile_result);
131 DeviceMemory<T> *c, int ldc, blas::ProfileResult *output_profile_result);
140 blas::ProfileResult *output_profile_result);
|
| D | cuda_blas.cc | 1899 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 …]
|
| D | cuda_dnn.cc | 1762 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/ |
| D | dnn.h | 774 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 …]
|
| D | stream.h | 71 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 …]
|
| D | blas.h | 163 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 …]
|
| D | stream.cc | 458 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 …]
|
| D | stream_executor_pimpl.h | 379 std::vector<dnn::ProfileResult> *out_algorithms);
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.h | 85 se::blas::ProfileResult* profile_result = nullptr,
|
| D | gpu_conv_algorithm_picker.cc | 124 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()
|
| D | gpu_conv_runner.h | 36 se::dnn::ProfileResult* profile_result = nullptr;
|
| D | gemm_thunk.cc | 86 se::blas::ProfileResult *output_profile_result) { in DoGemmWithAlgorithm()
177 se::blas::ProfileResult *profile_result, in RunGemm()
|
| D | gemm_algorithm_picker.cc | 128 se::blas::ProfileResult profile_result; in DoUncachedGemmAutotune()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 74 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 ()()
|
| D | conv_ops_3d.cc | 497 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()
|
| D | conv_grad_ops_3d.cc | 1417 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()
|
| D | cudnn_rnn_ops.cc | 126 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()
|
| D | conv_grad_filter_ops.cc | 658 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 ()()
|
| D | conv_ops.cc | 654 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 ()()
|
| D | conv_ops_fused_impl.h | 403 se::dnn::ProfileResult profile_result;
672 se::dnn::ProfileResult* profile_result) -> Status {
|