| /external/tensorflow/tensorflow/core/kernels/ |
| matmul_op.cc | 296 AlgorithmConfig algorithm_config(kNoAlgorithm);
305 &algorithm_config)) {
365 algorithm_config.set_algorithm(best_result.algorithm());
368 algorithm_config);
369 if (algorithm_config.algorithm() != kNoAlgorithm &&
370 algorithm_config.algorithm() != kDefaultBlasGemm &&
371 algorithm_config.algorithm() != kDefaultBlasGemv) {
377 &c_ptr, n, computation_type, algorithm_config.algorithm(),
397 // algorithm_config.algorithm() to be kDefaultBlasGemv.
399 algorithm_config.algorithm() == kNoAlgorithm |
[all...] |
| conv_ops_3d.cc | 393 AlgorithmConfig algorithm_config;
396 conv_parameters, &algorithm_config)) {
432 algorithm_config.set_algorithm(best_result.algorithm());
435 algorithm_config.set_algorithm_no_scratch(
438 AutoTuneConv3d::GetInstance()->Insert(conv_parameters, algorithm_config);
447 algorithm_config, nullptr)
|
| conv_grad_ops_3d.cc | [all...] |
| 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...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| convolution_thunk.cc | 75 se::dnn::AlgorithmConfig algorithm_config(
80 filter_data, output_data, scratch, window_, dim_nums_, algorithm_config,
|
| convolution_thunk.h | 83 const perftools::gputools::dnn::AlgorithmConfig& algorithm_config,
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.h | 253 const dnn::AlgorithmConfig& algorithm_config,
272 const dnn::AlgorithmConfig& algorithm_config,
286 const dnn::AlgorithmConfig& algorithm_config,
300 const dnn::AlgorithmConfig& algorithm_config,
318 const dnn::AlgorithmConfig& algorithm_config,
332 const dnn::AlgorithmConfig& algorithm_config,
383 const dnn::AlgorithmConfig& algorithm_config,
395 const dnn::AlgorithmConfig& algorithm_config,
407 const dnn::AlgorithmConfig& algorithm_config,
419 const dnn::AlgorithmConfig& algorithm_config,
[all...] |
| cuda_dnn.cc | [all...] |
| /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/stream_executor/ |
| dnn.h | [all...] |
| stream.h | 369 const dnn::AlgorithmConfig &algorithm_config,
380 const dnn::AlgorithmConfig &algorithm_config,
394 const dnn::AlgorithmConfig &algorithm_config,
408 const dnn::AlgorithmConfig &algorithm_config,
423 const dnn::AlgorithmConfig &algorithm_config,
437 const dnn::AlgorithmConfig &algorithm_config,
488 const dnn::AlgorithmConfig &algorithm_config,
500 const dnn::AlgorithmConfig &algorithm_config,
541 const dnn::AlgorithmConfig &algorithm_config,
553 const dnn::AlgorithmConfig &algorithm_config,
[all...] |
| stream.cc | 580 const dnn::AlgorithmConfig &algorithm_config,
587 PARAM(algorithm_config));
596 algorithm_config, output_profile_result);
619 const dnn::AlgorithmConfig &algorithm_config,
626 PARAM(output_descriptor), PARAM(output), PARAM(algorithm_config));
635 algorithm_config, output_profile_result);
657 const dnn::AlgorithmConfig &algorithm_config,
664 PARAM(output_descriptor), PARAM(output), PARAM(algorithm_config));
673 algorithm_config, output_profile_result);
692 const dnn::AlgorithmConfig &algorithm_config,
[all...] |
