Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_algorithm_picker.cc259 result.mutable_conv()->set_algorithm(alg.algo_id()); in PickBestAlgorithm()
292 failure->set_algorithm(first_algorithm->algo_id()); in PickBestAlgorithm()
385 backend_config.set_algorithm(best_algo.conv().algorithm()); in RunOnInstruction()
/external/tensorflow/tensorflow/core/kernels/
Dbias_op_gpu.h69 void set_algorithm(BiasAddGradGPUMode val) { algorithm_ = val; } in set_algorithm() function
Dgpu_utils.cc139 algo->set_algorithm({best_result->conv().algorithm(), in BestCudnnConvAlgorithm()
Dbias_op.cc578 best_result.set_algorithm(BiasAddGradGPUMode::kNative); in Compute()
592 best_result.set_algorithm(BiasAddGradGPUMode::kReduction); in Compute()
Dcudnn_rnn_ops.cc1314 output_algo_config->set_algorithm(algo_desc); in ComputeAndReturnAlgorithm()
1570 best_result.set_algorithm(algo); in MaybeAutoTune()
1577 algo_config->set_algorithm(best_result.algorithm()); in MaybeAutoTune()
1852 algo_config->set_algorithm(algo_desc); in GetAlgorithm()
Dconv_ops_3d.cc467 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in launch()
Dmatmul_op.cc360 algorithm_config.set_algorithm(best_result.algorithm()); in launch()
Dconv_grad_ops_3d.cc1398 algorithm_config.set_algorithm(best_result.algorithm()); in Compute()
1805 algorithm_config.set_algorithm(best_result.algorithm()); in Compute()
Dconv_ops_fused_impl.h536 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id());
Dconv_ops.cc877 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
Dconv_grad_filter_ops.cc865 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
Dconv_grad_input_ops.cc976 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc372 algo->set_algorithm({best_result->conv().algorithm(), in BestCudnnConvAlgorithm()
733 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in launch()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h776 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; }
805 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; }
Dblas.h137 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm() function
152 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm() function
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1650 output_profile_result->set_algorithm(algo_desc); in DoRnnForwardImpl()
1807 output_profile_result->set_algorithm(algo_desc); in DoRnnBackwardImpl()
2912 output_profile_result->set_algorithm(algorithm_desc); in DoConvolve()
3029 output_profile_result->set_algorithm(algo_desc); in DoFusedConvolveImpl()
Dcuda_blas.cc1823 output_profile_result->set_algorithm(blas::kDefaultBlasGemv); in DoBlasGemvWithProfilingImpl()
1855 output_profile_result->set_algorithm(blas::kDefaultBlasGemm); in DoBlasGemmWithProfilingImpl()
1974 output_profile_result->set_algorithm(algorithm); in DoBlasGemmWithAlgorithmImpl()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2931 output_profile_result->set_algorithm(algotype); in DoConvolve()