Searched defs:conv_parameters (Results 1 - 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_grad_ops_3d.cc | 642 const ConvParameters conv_parameters = { variable
662 conv_parameters, &algorithm_config)) {
665 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(), &algorithms));
705 AutoTuneConv3dBwdData::GetInstance()->Insert(conv_parameters,
1011 const ConvParameters conv_parameters = { variable
1029 conv_parameters, &algorithm_config)) {
1032 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(), &algorithms));
1073 AutoTuneConv3dBwdFilter::GetInstance()->Insert(conv_parameters,
|
| H A D | conv_ops_3d.cc | 374 ConvParameters conv_parameters = { local
396 conv_parameters, &algorithm_config)) {
399 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(), &algorithms));
438 AutoTuneConv3d::GetInstance()->Insert(conv_parameters, algorithm_config);
|
| H A D | conv_grad_filter_ops.cc | 890 ConvParameters conv_parameters = { local
909 conv_parameters, &algorithm_config)) {
912 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(), &algorithms));
953 AutoTuneConvBwdFilter::GetInstance()->Insert(conv_parameters,
|
| H A D | conv_grad_input_ops.cc | 939 ConvParameters conv_parameters = { local
958 conv_parameters, &algorithm_config)) {
961 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(), &algorithms));
1001 AutoTuneConvBwdData::GetInstance()->Insert(conv_parameters,
|
| H A D | conv_ops.cc | 690 ConvParameters conv_parameters = { local
709 !AutoTuneConv::GetInstance()->Find(conv_parameters, &algorithm_config)) {
712 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(), &algorithms));
753 AutoTuneConv::GetInstance()->Insert(conv_parameters, algorithm_config);
|
Completed in 90 milliseconds