| /external/tensorflow/tensorflow/python/eager/ |
| D | profiler_test.py | 42 profile_result = profiler.stop()
44 profile_pb.ParseFromString(profile_result)
53 profile_result = profile_pb.SerializeToString()
54 profiler.save(logdir, profile_result)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op.cc | 301 ProfileResult profile_result; in launch() local
314 &profile_result) in launch()
317 if (profile_result.is_valid()) { in launch()
318 if (profile_result.elapsed_time_in_ms() < in launch()
320 best_result = profile_result; in launch()
331 &c_ptr, n, &profile_result) in launch()
334 if (profile_result.is_valid()) { in launch()
335 if (profile_result.elapsed_time_in_ms() < in launch()
337 best_result = profile_result; in launch()
345 a_ptr, b_ptr, &c_ptr, &profile_result); in launch()
[all …]
|
| D | conv_grad_ops_3d.cc | 1372 ProfileResult profile_result; in Compute() local
1378 AlgorithmConfig(profile_algorithm), &profile_result) in Compute()
1381 if (profile_result.is_valid()) { in Compute()
1382 if (profile_result.elapsed_time_in_ms() < in Compute()
1384 best_result = profile_result; in Compute()
1387 profile_result.elapsed_time_in_ms() < in Compute()
1389 best_result_no_scratch = profile_result; in Compute()
1778 ProfileResult profile_result; in Compute() local
1785 &profile_result) in Compute()
1788 if (profile_result.is_valid()) { in Compute()
[all …]
|
| D | conv_ops_fused_impl.h | 527 se::dnn::ProfileResult profile_result;
531 &profile_result);
533 if (cudnn_launch_status && profile_result.is_valid()) {
543 absl::Milliseconds(profile_result.elapsed_time_in_ms()));
778 se::dnn::ProfileResult* profile_result) -> bool {
789 scratch_allocator, algorithm_config, profile_result)
|
| D | conv_ops_3d.cc | 455 ProfileResult profile_result; in launch() local
461 AlgorithmConfig(profile_algorithm), &profile_result) in launch()
464 if (profile_result.is_valid()) { in launch()
474 absl::Milliseconds(profile_result.elapsed_time_in_ms())); in launch()
|
| D | conv_ops.cc | 865 ProfileResult profile_result; in operator ()() local
871 AlgorithmConfig(profile_algorithm), &profile_result) in operator ()()
874 if (profile_result.is_valid()) { in operator ()()
884 absl::Milliseconds(profile_result.elapsed_time_in_ms())); in operator ()()
|
| D | conv_grad_filter_ops.cc | 852 ProfileResult profile_result; in operator ()() local
859 &profile_result) in operator ()()
862 if (profile_result.is_valid()) { in operator ()()
872 absl::Milliseconds(profile_result.elapsed_time_in_ms())); in operator ()()
|
| D | conv_grad_input_ops.cc | 964 ProfileResult profile_result; in operator ()() local
970 AlgorithmConfig(profile_algorithm), &profile_result) in operator ()()
973 if (profile_result.is_valid()) { in operator ()()
983 absl::Milliseconds(profile_result.elapsed_time_in_ms())); in operator ()()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_runner.cc | 116 se::dnn::ProfileResult* profile_result) { in RunCudnnConvImpl() argument
248 scratch_allocator, algorithm, profile_result); in RunCudnnConvImpl()
259 scratch_allocator, algorithm, profile_result); in RunCudnnConvImpl()
270 scratch_allocator, algorithm, profile_result); in RunCudnnConvImpl()
304 profile_result); in RunCudnnConvImpl()
421 options.profile_result); in RunCudnnConv()
424 options.profile_result); in RunCudnnConv()
427 options.profile_result); in RunCudnnConv()
|
| D | gemm_thunk.cc | 166 se::blas::ProfileResult profile_result; in DoGemmAutotune() local
173 stream, &profile_result)); in DoGemmAutotune()
175 if (profile_result.is_valid()) { in DoGemmAutotune()
177 << profile_result.elapsed_time_in_ms() << "ms"; in DoGemmAutotune()
178 if (profile_result.elapsed_time_in_ms() < in DoGemmAutotune()
180 best_result = profile_result; in DoGemmAutotune()
|
| D | cudnn_conv_algorithm_picker.cc | 237 se::dnn::ProfileResult profile_result; in PickBestAlgorithm() local
243 options.profile_result = &profile_result; in PickBestAlgorithm()
253 if (!profile_result.is_valid()) { in PickBestAlgorithm()
266 absl::Milliseconds(profile_result.elapsed_time_in_ms())); in PickBestAlgorithm()
|
| D | cudnn_conv_runner.h | 33 se::dnn::ProfileResult* profile_result = nullptr; member
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 719 dnn::ProfileResult profile_result; in launch() local
727 dnn::AlgorithmConfig(profile_algorithm), &profile_result) in launch()
730 if (profile_result.is_valid()) { in launch()
740 absl::Milliseconds(profile_result.elapsed_time_in_ms())); in launch()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 3600 Args... args, blas::ProfileResult *profile_result) { in operator ()()
3602 bool record_error = profile_result == nullptr; in operator ()()
3603 return Runner.Run(stream, blas_func, record_error, args..., profile_result); in operator ()()
|