Home
last modified time | relevance | path

Searched refs:ProfileResult (Results 1 – 25 of 25) 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_blas.cc1583 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1593 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1604 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1615 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1626 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1636 int ldc, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1647 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1659 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1671 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1682 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfilingImpl()
[all …]
Drocm_dnn.cc2341 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2383 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2426 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2455 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2506 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2558 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2817 dnn::ProfileResult* output_profile_result) { in DoConvolve()
3157 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve()
3174 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve()
3192 dnn::ProfileResult* output_profile_result) { in DoFusedConvolve()
[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_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()
1714 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1725 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1736 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfiling()
1747 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1757 int ldc, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1768 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1780 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1792 blas::ProfileResult *output_profile_result) { in DoBlasGemmWithProfiling()
1803 blas::ProfileResult *output_profile_result) { in DoBlasGemvWithProfilingImpl()
[all …]
Dcuda_dnn.cc1522 dnn::ProfileResult* output_profile_result) { in DoRnnForwardImpl()
1682 dnn::ProfileResult* output_profile_result) { in DoRnnBackwardImpl()
1888 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
1929 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
1971 dnn::ProfileResult* output_profile_result) { in DoRnnForward()
2020 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2071 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2123 dnn::ProfileResult* output_profile_result) { in DoRnnBackward()
2691 dnn::ProfileResult* output_profile_result) { in DoConvolve()
2935 dnn::ProfileResult* output_profile_result) { in DoFusedConvolveImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h768 class ProfileResult {
1127 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) {
1351 ProfileResult* output_profile_result) {
1398 ProfileResult* output_profile_result) {
2133 dnn::ProfileResult* output_profile_result) {
[all …]
Dstream.h70 class ProfileResult; variable
313 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);
[all …]
Dblas.h132 class ProfileResult {
484 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;
[all …]
Dstream.cc452 dnn::ProfileResult *output_profile_result) { in ThenFusedConvolveWithAlgorithm()
490 dnn::ProfileResult *output_profile_result) { in ThenFusedConvolveWithAlgorithm()
529 dnn::ProfileResult *output_profile_result) { in ThenFusedConvolveWithAlgorithm()
567 dnn::ProfileResult *output_profile_result) { in ThenFusedConvolveWithAlgorithm()
602 dnn::ProfileResult *output_profile_result) { in ThenConvolveWithAlgorithm()
644 dnn::ProfileResult *output_profile_result) { in ThenConvolveWithAlgorithm()
686 dnn::ProfileResult *output_profile_result) { in ThenConvolveWithAlgorithm()
831 dnn::ProfileResult *output_profile_result) { in ThenConvolveBackwardDataWithAlgorithm()
876 dnn::ProfileResult *output_profile_result) { in ThenConvolveBackwardDataWithAlgorithm()
921 dnn::ProfileResult *output_profile_result) { in ThenConvolveBackwardDataWithAlgorithm()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op.cc166 se::blas::ProfileResult* output_profile) { in Compute()
201 se::blas::ProfileResult* output_profile) { in Compute()
263 using se::blas::ProfileResult; in launch()
297 ProfileResult best_result; in launch()
301 ProfileResult profile_result; in launch()
Dconv_grad_ops_3d.cc1356 using se::dnn::ProfileResult; in Compute()
1365 ProfileResult best_result; in Compute()
1366 ProfileResult best_result_no_scratch; in Compute()
1372 ProfileResult profile_result; in Compute()
1762 using se::dnn::ProfileResult; in Compute()
1771 ProfileResult best_result; in Compute()
1772 ProfileResult best_result_no_scratch; in Compute()
1778 ProfileResult profile_result; in Compute()
Dcudnn_rnn_ops.cc126 using se::dnn::ProfileResult;
740 ProfileResult* output_profile_result) { in DoForward()
802 ProfileResult* output_profile_result) { in DoBackward()
1511 ProfileResult best_result; in MaybeAutoTune()
1516 ProfileResult final_profile_result; in MaybeAutoTune()
1518 ProfileResult fwd_profile_result; in MaybeAutoTune()
1519 ProfileResult bak_profile_result; in MaybeAutoTune()
Dconv_ops_3d.cc433 using se::dnn::ProfileResult; in launch()
455 ProfileResult profile_result; in launch()
Dconv_ops_fused_impl.h527 se::dnn::ProfileResult profile_result;
778 se::dnn::ProfileResult* profile_result) -> bool {
Dconv_ops.cc586 using se::dnn::ProfileResult; in operator ()()
865 ProfileResult profile_result; in operator ()()
Dconv_grad_filter_ops.cc551 using se::dnn::ProfileResult; in operator ()()
852 ProfileResult profile_result; in operator ()()
Dconv_grad_input_ops.cc688 using se::dnn::ProfileResult; in operator ()()
964 ProfileResult profile_result; in operator ()()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_runner.h33 se::dnn::ProfileResult* profile_result = nullptr;
Dgemm_thunk.cc120 se::blas::ProfileResult* output_profile_result) { in DoGemmWithAlgorithm()
164 se::blas::ProfileResult best_result; in DoGemmAutotune()
166 se::blas::ProfileResult profile_result; in DoGemmAutotune()
Dcudnn_conv_runner.cc40 using se::dnn::ProfileResult;
116 se::dnn::ProfileResult* profile_result) { in RunCudnnConvImpl()
Dcudnn_conv_algorithm_picker.cc237 se::dnn::ProfileResult profile_result; in PickBestAlgorithm()
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc719 dnn::ProfileResult profile_result; in launch()