Home
last modified time | relevance | path

Searched defs:is_grouped_convolution (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h195 bool* is_grouped_convolution, in GetFilterSizeInMklOrder()
301 bool* is_grouped_convolution, in GetFilterSizeInMklOrder()
515 memory::dims* pad_r, bool is_grouped_convolution, bool is_depthwise) { in GetOutputAndPadSizeInMklOrder()
Dmkl_conv_grad_input_ops.cc375 bool is_grouped_convolution = false; in Compute() local
Dmkl_conv_grad_filter_ops.cc423 bool is_grouped_convolution = false; in Compute() local
Dmkl_conv_ops.cc653 bool is_grouped_convolution = false; in Compute() local
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_3d.cc262 bool is_grouped_convolution = filter_depth != in_depth; in launch() local
Dconv_grad_input_ops.cc134 bool is_grouped_convolution = filter_shape.dim_size(2) != dims.in_depth; in operator ()() local
Dconv_grad_ops_3d.cc1281 bool is_grouped_convolution = filter_shape.dim_size(3) != dims.in_depth; in Compute() local
1826 bool is_grouped_convolution = filter_shape.dim_size(3) != dims.in_depth; in Compute() local
Dconv_grad_filter_ops.cc729 bool is_grouped_convolution = filter_shape.dim_size(2) != dims.in_depth; in operator ()() local
Dconv_ops.cc812 bool is_grouped_convolution = patch_depths != in_depths; in operator ()() local