Searched refs:conv_parameters (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 362 ConvParameters conv_parameters = { in operator ()() local
390 conv_parameters, &algorithm_config)) { in operator ()()
404 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(stream->parent()), in operator ()()
497 AutoTuneConvBwdData::GetInstance()->Insert(conv_parameters, in operator ()()
|
| D | conv_ops_3d.cc | 481 ConvParameters conv_parameters = { in launch() local
508 conv_parameters, &algorithm_config)) { in launch()
519 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>( in launch()
621 AutoTuneConv3d::GetInstance()->Insert(conv_parameters, algorithm_config); in launch()
|
| D | conv_grad_ops_3d.cc | 1401 const ConvParameters conv_parameters = { in Compute() local
1426 conv_parameters, &algorithm_config)) { in Compute()
1436 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>( in Compute()
1518 AutoTuneConv3dBwdData::GetInstance()->Insert(conv_parameters, in Compute()
1898 const ConvParameters conv_parameters = { in Compute() local
1923 conv_parameters, &algorithm_config)) { in Compute()
1927 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>( in Compute()
2011 AutoTuneConv3dBwdFilter::GetInstance()->Insert(conv_parameters, in Compute()
|
| D | conv_grad_filter_ops.cc | 957 ConvParameters conv_parameters = { in operator ()() local
985 conv_parameters, &algorithm_config)) { in operator ()()
998 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>(stream->parent()), in operator ()()
1096 AutoTuneConvBwdFilter::GetInstance()->Insert(conv_parameters, in operator ()()
|
| D | conv_ops.cc | 967 ConvParameters conv_parameters = {in_batch, // batch in operator ()() local
993 !AutoTuneConv::GetInstance()->Find(conv_parameters, &algorithm_config)) { in operator ()()
999 conv_parameters.ShouldIncludeWinogradNonfusedAlgo<T>( in operator ()()
1107 AutoTuneConv::GetInstance()->Insert(conv_parameters, algorithm_config); in operator ()()
|
| D | conv_ops_fused_impl.h | 645 FusedConvParameters conv_parameters = {
688 conv_parameters, launch, context, stream, output_ptr,
|