Searched refs:algo_config (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 1040 const AlgorithmConfig algo_config; in ExtractCudnnRNNParamsInfo() local
1044 ToDataType<T>::value, algo_config, dropout(), seed(), in ExtractCudnnRNNParamsInfo()
1057 const AlgorithmConfig& algo_config, in CreateRnnDescriptor() argument
1067 data_type, algo_config, dropout(), seed(), dropout_state_allocator, in CreateRnnDescriptor()
1084 const AlgorithmConfig& algo_config, in GetCachedRnnDescriptor() argument
1087 auto key = std::make_pair(model_shapes, algo_config.algorithm()); in GetCachedRnnDescriptor()
1094 context, model_shapes, input_mode, algo_config, in GetCachedRnnDescriptor()
1446 AlgorithmConfig algo_config; in Compute() local
1447 ComputeAndReturnAlgorithm(context, &algo_config, /*var_seq_lengths=*/false, in Compute()
1638 AlgorithmConfig* algo_config) override { in MaybeAutoTune() argument
[all …]
|
| D | bias_op.cc | 532 BiasAddGradGPUConfig algo_config; in Compute() local
533 if (!AutotuneBiasGrad::GetInstance()->Find(bias_parameters, &algo_config)) { in Compute()
564 algo_config.set_mode(best_result.algorithm()); in Compute()
565 AutotuneBiasGrad::GetInstance()->Insert(bias_parameters, algo_config); in Compute()
572 if (algo_config.get_mode() == BiasAddGradGPUMode::kReduction) { in Compute()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 61 std::string ToVlogString(const dnn::AlgorithmConfig &algo_config) { in ToVlogString() argument
62 return algo_config.ToString(); in ToVlogString()
|