Searched refs:conv_parameters (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_3d.cc | 417 ConvParameters conv_parameters = { in launch() local
438 conv_parameters, &algorithm_config)) { in launch()
442 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>( in launch()
481 AutoTuneConv3d::GetInstance()->Insert(conv_parameters, algorithm_config); in launch()
|
| D | conv_grad_ops_3d.cc | 1340 const ConvParameters conv_parameters = { in Compute() local
1359 conv_parameters, &algorithm_config)) { in Compute()
1362 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>( in Compute()
1404 AutoTuneConv3dBwdData::GetInstance()->Insert(conv_parameters, in Compute()
1746 const ConvParameters conv_parameters = { in Compute() local
1765 conv_parameters, &algorithm_config)) { in Compute()
1768 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>( in Compute()
1811 AutoTuneConv3dBwdFilter::GetInstance()->Insert(conv_parameters, in Compute()
|
| D | conv_ops.cc | 828 ConvParameters conv_parameters = { in operator ()() local
849 !AutoTuneConv::GetInstance()->Find(conv_parameters, &algorithm_config)) { in operator ()()
854 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>( in operator ()()
891 AutoTuneConv::GetInstance()->Insert(conv_parameters, algorithm_config); in operator ()()
|
| D | conv_grad_filter_ops.cc | 820 ConvParameters conv_parameters = { in operator ()() local
841 conv_parameters, &algorithm_config)) { in operator ()()
844 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(stream->parent()), in operator ()()
880 AutoTuneConvBwdFilter::GetInstance()->Insert(conv_parameters, in operator ()()
|
| D | conv_grad_input_ops.cc | 932 ConvParameters conv_parameters = { in operator ()() local
953 conv_parameters, &algorithm_config)) { in operator ()()
956 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(stream->parent()), in operator ()()
991 AutoTuneConvBwdData::GetInstance()->Insert(conv_parameters, in operator ()()
|
| D | conv_ops_fused_impl.h | 751 FusedConvParameters conv_parameters = {
796 conv_parameters, launch, context, stream,
|