OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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