Searched refs:elements_depth (Results 1 – 1 of 1) sorted by relevance
133 auto elements_depth = elements_shape.empty() ? 1 : elements_shape.back(); in CanFuseConvOrDepthwiseConvShapes() local136 if (elements_depth == 1) { in CanFuseConvOrDepthwiseConvShapes()146 (is_depthwise ? filter_shape.back() != elements_depth in CanFuseConvOrDepthwiseConvShapes()147 : filter_shape[0] != elements_depth)) in CanFuseConvOrDepthwiseConvShapes()