HomeSort by relevance Sort by last modified time
    Searched defs:algorithm_config (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/contrib/fused_conv/kernels/
fused_conv2d_bias_activation_op.cc 541 dnn::AlgorithmConfig algorithm_config; local
543 fused_conv_parameters, &algorithm_config)) {
583 algorithm_config.set_algorithm(best_result.algorithm());
586 algorithm_config.set_algorithm_no_scratch(
590 algorithm_config);
600 &output_ptr, &scratch_allocator, algorithm_config,
  /external/tensorflow/tensorflow/core/kernels/
conv_ops.cc 707 AlgorithmConfig algorithm_config; local
709 !AutoTuneConv::GetInstance()->Find(conv_parameters, &algorithm_config)) {
747 algorithm_config.set_algorithm(best_result.algorithm());
750 algorithm_config.set_algorithm_no_scratch(
753 AutoTuneConv::GetInstance()->Insert(conv_parameters, algorithm_config);
762 algorithm_config, nullptr)
    [all...]
conv_grad_filter_ops.cc 907 AlgorithmConfig algorithm_config; local
    [all...]
conv_grad_input_ops.cc 956 AlgorithmConfig algorithm_config; local
    [all...]

Completed in 103 milliseconds