Home
last modified time | relevance | path

Searched defs:profile_result (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.cc62 se::dnn::ProfileResult profile_result; in AutotuneConvImpl() local
146 se::dnn::ProfileResult* profile_result) -> Status { in AutotuneFusedConv()
325 se::dnn::ProfileResult* profile_result) -> Status { in AutotuneUnfusedConv()
399 auto profile_result = algorithms[0]; in AutotuneUnfusedConv() local
410 se::dnn::ProfileResult profile_result; in AutotuneUnfusedConv() local
Dmatmul_op_fused.cc197 se::blas::ProfileResult profile_result; in AutotuneMatmul() local
304 se::blas::ProfileResult* profile_result) { in operator ()()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dmatmul_utils.cc429 se::blas::ProfileResult* profile_result) { in DoGemmWithAlgorithm()
460 se::blas::ProfileResult* profile_result) { in DoGemm()
490 se::blas::ProfileResult* profile_result) { in RunGemm()
694 se::blas::ProfileResult* profile_result) { in DoMatmul()
723 se::blas::ProfileResult* profile_result) { in ExecuteOnStream()
Dgemm_algorithm_picker.cc298 se::blas::ProfileResult profile_result; in DoGemmAutotune() local
317 se::blas::ProfileResult profile_result; in DoGemmAutotune() local
Dgpu_conv_algorithm_picker.cc493 se::dnn::ProfileResult profile_result; in AutotuneOneConvRunner() local
860 se::dnn::ProfileResult profile_result; in PickBestAlgorithmNoCacheRocm() local
Dgpu_conv_runner.h157 se::dnn::ProfileResult* profile_result = nullptr; member
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_blas_lt.cc264 blas::ProfileResult* profile_result) { in DoMatmul()
Dcuda_dnn.cc4080 port::Status operator()(Stream* stream, dnn::ProfileResult* profile_result, in operator ()()
4265 dnn::ProfileResult* profile_result) { in DoConvolve()
4450 port::Status operator()(Stream* stream, dnn::ProfileResult* profile_result, in operator ()()
4918 port::Status operator()(Stream* stream, dnn::ProfileResult* profile_result, in operator ()()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc511 dnn::ProfileResult profile_result; in GetProfileResultFromConvSolution() local
537 dnn::ProfileResult profile_result; in GetProfileResultFromConvAlgoPerf() local
2981 port::Status operator()(Stream* stream, dnn::ProfileResult* profile_result, in operator ()()
3173 for (const auto& profile_result : profile_results) { in GetConvolveRunners() local