Home
last modified time | relevance | path

Searched refs:set_algorithm (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_algorithm_picker.cc461 result.mutable_conv()->set_algorithm(alg.algo_id()); in PickBestAlgorithmNoCacheCuda()
532 reference_conv->set_algorithm(first_algorithm.algo_id()); in PickBestAlgorithmNoCacheCuda()
642 result.mutable_conv()->set_algorithm(profile_result.algorithm().algo_id()); in PickBestAlgorithmNoCacheRocm()
681 result.mutable_conv()->set_algorithm(alg.algo_id()); in PickBestAlgorithmNoCacheRocm()
724 backend_config.set_algorithm(best_algo.conv().algorithm()); in RunOnInstruction()
Dgemm_algorithm_picker.cc147 result.mutable_gemm()->set_algorithm(algorithm); in DoUncachedGemmAutotune()
189 result.mutable_failure()->mutable_reference_gemm()->set_algorithm( in DoUncachedGemmAutotune()
Dir_emitter_unnested.cc981 descriptor.backend_config.set_algorithm( in EmitConvolutionThunk()
/external/tensorflow/tensorflow/core/kernels/
Dbias_op_gpu.h70 void set_algorithm(BiasAddGradGPUMode val) { algorithm_ = val; } in set_algorithm() function
Dgpu_utils.cc249 algo->set_algorithm({results[idx].conv().algorithm(), in BestCudnnConvAlgorithm()
262 algo->set_algorithm( in BestCudnnConvAlgorithm()
Dconv_grad_input_ops.cc466 result.mutable_conv()->set_algorithm( in operator ()()
513 result.mutable_conv()->set_algorithm( in operator ()()
534 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
Dconv_ops_3d.cc584 result.mutable_conv()->set_algorithm( in launch()
629 result.mutable_conv()->set_algorithm( in launch()
651 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in launch()
Dbias_op.cc550 best_result.set_algorithm(BiasAddGradGPUMode::kNative); in Compute()
564 best_result.set_algorithm(BiasAddGradGPUMode::kReduction); in Compute()
Dconv_grad_filter_ops.cc1072 result.mutable_conv()->set_algorithm( in operator ()()
1121 result.mutable_conv()->set_algorithm( in operator ()()
1142 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
Dconv_ops.cc1191 result.mutable_conv()->set_algorithm( in operator ()()
1240 result.mutable_conv()->set_algorithm( in operator ()()
1260 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
Dcudnn_rnn_ops.cc1540 output_algo_config->set_algorithm(algo_desc); in ComputeAndReturnAlgorithm()
1799 best_result.set_algorithm(algo); in MaybeAutotune()
1806 algo_config->set_algorithm(best_result.algorithm()); in MaybeAutotune()
2095 algo_config->set_algorithm(algo_desc); in GetAlgorithm()
Dconv_grad_ops_3d.cc1593 result.mutable_conv()->set_algorithm( in Compute()
1643 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in Compute()
2137 result.mutable_conv()->set_algorithm( in Compute()
2182 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in Compute()
Dconv_ops_fused_impl.h401 result.mutable_conv()->set_algorithm(
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1963 output_profile_result->set_algorithm(algo_desc); in DoRnnForwardImpl()
2073 output_profile_result->set_algorithm(algo_desc); in DoRnnForwardImpl()
2196 output_profile_result->set_algorithm(algo_desc); in DoRnnBackwardImpl()
2305 output_profile_result->set_algorithm(algo_desc); in DoRnnBackwardImpl()
4050 output_profile_result->set_algorithm(algorithm_desc); in DoConvolve()
4158 output_profile_result->set_algorithm(selected_plan_); in DoConvolveWithExecutionPlan()
5001 output_profile_result->set_algorithm(algo_desc); in DoFusedConvolve()
5128 output_profile_result->set_algorithm(selected_plan); in DoFusedConvolveWithExecutionPlan()
Dcuda_blas.cc2002 output_profile_result->set_algorithm(blas::kDefaultBlasGemv); in DoBlasInternalImpl()
2036 output_profile_result->set_algorithm(blas::kDefaultBlasGemm); in DoBlasInternalImpl()
2140 output_profile_result->set_algorithm(algorithm); in DoBlasInternalImpl()
3685 output_profile_result->set_algorithm(algorithm->index()); in DoBlasInternalImpl()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h836 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; } in set_algorithm() function
905 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; } in set_algorithm() function
Dblas.h199 void set_algorithm(AlgorithmType val) { algorithm_ = val; }
214 void set_algorithm(AlgorithmType val) { algorithm_ = val; }
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc508 profile_result.set_algorithm( in GetProfileResultFromConvSolution()
520 profile_result.set_algorithm( in GetProfileResultFromConvAlgoPerf()
524 profile_result.set_algorithm( in GetProfileResultFromConvAlgoPerf()
528 profile_result.set_algorithm( in GetProfileResultFromConvAlgoPerf()
3060 output_profile_result->set_algorithm(algotype); in DoConvolve()
/external/tensorflow/tensorflow/core/framework/
Dmodel_test.cc964 optimization_params.set_algorithm(AutotuneAlgorithm::GRADIENT_DESCENT); in TEST()
Dmodel.cc1657 optimization_params.set_algorithm(algorithm); in Optimize()