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

  /external/tensorflow/tensorflow/core/kernels/
bias_op_gpu.h 69 void set_algorithm(BiasAddGradGPUMode val) { algorithm_ = val; } function in class:tensorflow::BiasGradGPUProfileResult
gpu_utils.cc 139 algo->set_algorithm({best_result->conv().algorithm(),
bias_op.cc 578 best_result.set_algorithm(BiasAddGradGPUMode::kNative);
592 best_result.set_algorithm(BiasAddGradGPUMode::kReduction);
conv_ops_3d.cc 467 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id());
cudnn_rnn_ops.cc     [all...]
conv_ops_fused_impl.h 536 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id());
    [all...]
matmul_op.cc 360 algorithm_config.set_algorithm(best_result.algorithm());
conv_grad_ops_3d.cc     [all...]
conv_grad_filter_ops.cc     [all...]
conv_grad_input_ops.cc     [all...]
conv_ops.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_conv_algorithm_picker.cc 259 result.mutable_conv()->set_algorithm(alg.algo_id());
292 failure->set_algorithm(first_algorithm->algo_id());
385 backend_config.set_algorithm(best_algo.conv().algorithm());
  /external/tensorflow/tensorflow/contrib/fused_conv/kernels/
fused_conv2d_bias_activation_op.cc 372 algo->set_algorithm({best_result->conv().algorithm(),
733 result.mutable_conv()->set_algorithm(profile_algorithm.algo_id());
    [all...]
  /external/tensorflow/tensorflow/stream_executor/
dnn.h 776 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; } function in class:stream_executor::dnn::ProfileResult
805 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; } function in class:stream_executor::dnn::AlgorithmConfig
    [all...]
blas.h 137 void set_algorithm(AlgorithmType val) { algorithm_ = val; } function in class:stream_executor::blas::ProfileResult
152 void set_algorithm(AlgorithmType val) { algorithm_ = val; } function in class:stream_executor::blas::AlgorithmConfig
    [all...]
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.cc     [all...]
cuda_blas.cc     [all...]
  /external/tensorflow/tensorflow/stream_executor/rocm/
rocm_dnn.cc     [all...]

Completed in 494 milliseconds