Home
last modified time | relevance | path

Searched refs:set_group_count (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc401 config.conv_desc.set_group_count(desc.feature_group_count); in GetGpuConvConfig()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc490 proto_.set_group_count(1); in ConvolutionDescriptor()
Ddnn.h549 ConvolutionDescriptor& set_group_count(int group_count) { in set_group_count() function
550 proto_.set_group_count(group_count); in set_group_count()
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc274 .set_group_count(dims.in_depth / filter_shape.dim_size(2)); in operator ()()
Dconv_ops_3d.cc428 .set_group_count(in_depth / filter_depth); in launch()
Dconv_ops_fused_impl.h596 .set_group_count(in_depths / patch_depths);
Dconv_grad_ops_3d.cc1329 .set_group_count(dims.in_depth / filter_shape.dim_size(3)); in Compute()
1826 .set_group_count(dims.in_depth / filter_shape.dim_size(3)); in Compute()
Dconv_grad_filter_ops.cc879 .set_group_count(dims.in_depth / filter_shape.dim_size(2)); in operator ()()
Dconv_ops.cc904 .set_group_count(in_depths / patch_depths); in operator ()()