Searched refs:algo_config (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 963 const AlgorithmConfig algo_config; in ExtractCudnnRNNParamsInfo() local
966 rnn_direction_mode(), rnn_mode(), ToDataType<T>::value, algo_config, in ExtractCudnnRNNParamsInfo()
979 const AlgorithmConfig& algo_config, in CreateRnnDescriptor() argument
987 rnn_direction_mode(), rnn_mode(), data_type, algo_config, dropout(), in CreateRnnDescriptor()
1004 const AlgorithmConfig& algo_config, in GetCachedRnnDescriptor() argument
1007 auto key = std::make_pair(model_shapes, algo_config.algorithm()); in GetCachedRnnDescriptor()
1014 CreateRnnDescriptor<T>(context, model_shapes, input_mode, algo_config, in GetCachedRnnDescriptor()
1265 AlgorithmConfig algo_config; in Compute() local
1266 ComputeAndReturnAlgorithm(context, &algo_config, /*var_seq_lengths=*/false, in Compute()
1452 AlgorithmConfig* algo_config) override { in MaybeAutoTune() argument
[all …]
|
| D | bias_op.cc | 564 BiasAddGradGPUConfig algo_config; in Compute() local
565 if (!AutotuneBiasGrad::GetInstance()->Find(bias_parameters, &algo_config)) { in Compute()
596 algo_config.set_mode(best_result.algorithm()); in Compute()
597 AutotuneBiasGrad::GetInstance()->Insert(bias_parameters, algo_config); in Compute()
604 if (algo_config.get_mode() == BiasAddGradGPUMode::kReduction) { in Compute()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 62 string ToVlogString(const dnn::AlgorithmConfig &algo_config) { in ToVlogString() argument
63 return algo_config.ToString(); in ToVlogString()
|