Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc105 const bool is_depthwise_conv = input_channels == feature_group_count; in matchAndRewrite() local
130 CreateConvOp(conv_op, strides, padding, dilation, is_depthwise_conv, in matchAndRewrite()
160 bool is_depthwise_conv, int input_channels, in CreateConvOp() argument
163 if (is_depthwise_conv) { in CreateConvOp()