Searched refs:algo_config (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 1082 const AlgorithmConfig algo_config; in ExtractCudnnRNNParamsInfo() local
1086 ToDataType<T>::value, algo_config, dropout(), seed(), in ExtractCudnnRNNParamsInfo()
1099 const AlgorithmConfig& algo_config, in CreateRnnDescriptor() argument
1109 data_type, algo_config, dropout(), seed(), dropout_state_allocator, in CreateRnnDescriptor()
1126 const AlgorithmConfig& algo_config, in GetCachedRnnDescriptor() argument
1129 auto key = std::make_pair(model_shapes, algo_config.algorithm()); in GetCachedRnnDescriptor()
1136 context, model_shapes, input_mode, algo_config, in GetCachedRnnDescriptor()
1488 AlgorithmConfig algo_config; in Compute() local
1489 ComputeAndReturnAlgorithm(context, &algo_config, /*var_seq_lengths=*/false, in Compute()
1681 AlgorithmConfig* algo_config) override { in MaybeAutotune() argument
[all …]
|
| D | bias_op.cc | 479 BiasAddGradGPUConfig algo_config; in Compute() local
480 if (!AutotuneBiasGrad::GetInstance()->Find(bias_parameters, &algo_config)) { in Compute()
513 algo_config.set_mode(best_result.algorithm()); in Compute()
514 AutotuneBiasGrad::GetInstance()->Insert(bias_parameters, algo_config); in Compute()
521 if (algo_config.get_mode() == BiasAddGradGPUMode::kReduction) { in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream.cc | 63 std::string ToVlogString(const dnn::AlgorithmConfig &algo_config) { in ToVlogString() argument
64 return algo_config.ToString(); in ToVlogString()
|