Searched refs:algo_id (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 26 return ::tensorflow::Hash64Combine(algo_id(), tensor_ops_enabled()); in hash()
225 AlgorithmDesc::Index algo_id = -1; in ToString() local
227 algo_id = algorithm()->algo_id(); in ToString()
231 algo_id_no_scratch = algorithm_no_scratch()->algo_id(); in ToString()
233 return absl::StrCat(algo_id, ", ", algo_id_no_scratch); in ToString()
|
| D | dnn.proto | 94 int64 algo_id = 1; field
|
| D | dnn.h | 750 Index algo_id() const { return proto_.algo_id(); }
752 return algo_id() == other.algo_id() &&
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_algorithm_picker.cc | 68 return absl::StrCat(algo.algo_id(), "+TC"); in AlgorithmToString()
70 return absl::StrCat(algo.algo_id()); in AlgorithmToString()
259 result.mutable_conv()->set_algorithm(alg.algo_id()); in PickBestAlgorithm()
292 failure->set_algorithm(first_algorithm->algo_id()); in PickBestAlgorithm()
|
| D | cudnn_conv_runner.cc | 129 VLOG(3) << "Convolution Algorithm: " << algorithm.algorithm()->algo_id(); in RunCudnnConvImpl()
312 CudnnConvKindToString(kind), algorithm.algorithm()->algo_id(), in RunCudnnConvImpl()
313 algorithm.algorithm_no_scratch()->algo_id()); in RunCudnnConvImpl()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 209 cudnnConvolutionFwdAlgo_t(algorithm.algo_id()); in ToConvForwardAlgo()
222 << algorithm.algo_id(); in ToConvForwardAlgo()
229 cudnnConvolutionBwdDataAlgo_t(algorithm.algo_id()); in ToConvBackwardDataAlgo()
241 << algorithm.algo_id(); in ToConvBackwardDataAlgo()
248 cudnnConvolutionBwdFilterAlgo_t(algorithm.algo_id()); in ToConvBackwardFilterAlgo()
264 << algorithm.algo_id(); in ToConvBackwardFilterAlgo()
278 cudnnRNNAlgo_t algo = static_cast<cudnnRNNAlgo_t>(algorithm->algo_id()); in ToCudnnRNNAlgo()
285 LOG(FATAL) << "Unsupported Cudnn RNN algorithm: " << algorithm->algo_id(); in ToCudnnRNNAlgo()
2729 if (algorithm_desc.algo_id() != CUDNN_CONVOLUTION_FWD_ALGO_FFT_TILING) { in DoConvolve()
2752 if (algorithm_desc.algo_id() == in DoConvolve()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 1436 output_host_reserved_int8(0) = best_algo_config.algorithm()->algo_id(); in Compute()
1478 << algo_config->algorithm()->algo_id() << ", " in MaybeAutoTune()
1514 << algo.algo_id() << ", " << algo.tensor_ops_enabled() << ")."; in MaybeAutoTune()
1566 << algo.algo_id() << ", " << algo.tensor_ops_enabled() << ")" in MaybeAutoTune()
1579 << best_result.algorithm().algo_id() << ", " in MaybeAutoTune()
|
| D | conv_ops_3d.cc | 467 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in launch()
|
| D | conv_ops_fused_impl.h | 536 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id());
|
| D | conv_ops.cc | 877 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
|
| D | conv_grad_filter_ops.cc | 865 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
|
| D | conv_grad_input_ops.cc | 976 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in operator ()()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 394 miopenConvFwdAlgorithm_t algo = miopenConvFwdAlgorithm_t(algorithm.algo_id()); in ToConvForwardAlgo()
403 << algorithm.algo_id(); in ToConvForwardAlgo()
410 miopenConvBwdDataAlgorithm_t(algorithm.algo_id()); in ToConvBackwardDataAlgo()
420 << algorithm.algo_id(); in ToConvBackwardDataAlgo()
427 miopenConvBwdWeightsAlgorithm_t(algorithm.algo_id()); in ToConvBackwardFilterAlgo()
435 << algorithm.algo_id(); in ToConvBackwardFilterAlgo()
2859 static_cast<miopenConvFwdAlgorithm_t>(algorithm_desc.algo_id()), in DoConvolve()
2884 static_cast<miopenConvBwdDataAlgorithm_t>(algorithm_desc.algo_id()), in DoConvolve()
2911 algorithm_desc.algo_id()), in DoConvolve()
2930 dnn::AlgorithmDesc algotype(algorithm_desc.algo_id(), false); in DoConvolve()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 709 return alg.algo_id() != in launch()
733 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id()); in launch()
|