- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for divisor_override (2.48 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool_patterns.td
[(IsBoolCompositeAttribute<"ceil_mode", "false"> $attrs), (IsBoolCompositeAttribute<"count_include_pad", "false"> $attrs), (IsStrCompositeAttribute<"divisor_override", "py_None"> $attrs), (HasSameOrValidPadding $old_val)]>; // Replaces an aten.avg_pool2d with (T -> tfl.pad -> tfl.average_pool_2d -> T). def LegalizeAvgPool2dWithPadComposite: Pat<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/composite-lowering.mlir
// CHECK: } func.func @avg_pool2d_1(%arg0: tensor<1x3x6x6xf32>) -> (tensor<*xf32>) { %0 = mhlo.composite "aten.avg_pool2d.default" %arg0 {composite_attributes = {ceil_mode = false, count_include_pad = true, divisor_override = "py_None", kernel_size = dense<3> : tensor<2xi64>, padding = dense<0> : tensor<2xi64>, stride = dense<1> : tensor<2xi64>}, decomposition = @XlaCallModule_aten.avg_pool2d.default.impl_0} : (tensor<1x3x6x6xf32>) -> tensor<1x3x4x4xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 18:45:51 UTC 2024 - 32.6K bytes - Viewed (0)