HomeSort by relevance Sort by last modified time
    Searched refs:algo_config (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
cudnn_rnn_ops.cc 963 const AlgorithmConfig algo_config; local
1265 AlgorithmConfig algo_config; variable
1709 AlgorithmConfig algo_config; local
    [all...]
bias_op.cc 564 BiasAddGradGPUConfig algo_config; variable
565 if (!AutotuneBiasGrad::GetInstance()->Find(bias_parameters, &algo_config)) {
596 algo_config.set_mode(best_result.algorithm());
597 AutotuneBiasGrad::GetInstance()->Insert(bias_parameters, algo_config);
604 if (algo_config.get_mode() == BiasAddGradGPUMode::kReduction) {
  /external/tensorflow/tensorflow/stream_executor/
stream.cc 62 string ToVlogString(const dnn::AlgorithmConfig &algo_config) {
63 return algo_config.ToString();
    [all...]

Completed in 671 milliseconds