- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for CanFuseConvOrDepthwiseConv (0.51 sec)
-
tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td
defm : FuseActFnIntoPoolOpPat<!cast<Op>(actFnPair[0]), !cast<ConstantStrAttr>(actFnPair[1])>; } class CanFuseConvOrDepthwiseConv<string is_depthwise> : Constraint< CPred<"TFL::CanFuseConvOrDepthwiseConv($0, $1, " # is_depthwise # ")">>; // If we see a binary op (add, sub) op adding a constant value to a convolution
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 66.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
if (filter_shape.empty() || (is_depthwise ? filter_shape.back() != elements_depth : filter_shape[0] != elements_depth)) return false; return true; } bool CanFuseConvOrDepthwiseConv(Value filter, Attribute val, bool is_depthwise) { const auto elements = mlir::dyn_cast<DenseElementsAttr>(val); if (!elements) { return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0)