- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for filter_height (0.48 sec)
-
tensorflow/compiler/mlir/tfr/examples/mnist/ops_defs.py
attrs=[ 'stride_w: int', 'stride_h: int', 'filter_width: int', 'filter_height: int', 'padding: {"SAME", "VALID"}' ], derived_attrs=['T: {float, int8}'], outputs=['o: T']) def _composite_max_pool(input_, stride_w, stride_h, filter_width, filter_height, padding): ksize = [1, filter_width, filter_height, 1] strides = [1, stride_w, stride_h, 1] return tf.raw_ops.MaxPool(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 31 20:23:51 UTC 2023 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/tests/device-transform-nnapi.mlir
func.return %0 : tensor<1x1x1x512xf32> } // CHECK: func @mean_4d_keepdim([[VAL_0:%.*]]: tensor<1x48x48x512xf32>) -> tensor<1x1x1x512xf32> { // CHECK: [[VAL_1:%.*]] = "tfl.average_pool_2d"([[VAL_0]]) <{filter_height = 48 : i32, filter_width = 48 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 1 : i32, stride_w = 1 : i32}> : (tensor<1x48x48x512xf32>) -> tensor<1x1x1x512xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 4.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/many_attribute_op.mlir
^bb0(%arg0: tensor<1x6x6x16xf32>): // CHECK: "tfl.average_pool_2d"(%{{.*}}) <{filter_height = 3 : i32, filter_width = 6 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 3 : i32, stride_w = 1 : i32}> : (tensor<1x6x6x16xf32>) -> tensor<1x1x1x16xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 824 bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/nn.mlir
// CHECK-NEXT: padding: VALID, // CHECK-NEXT: stride_w: 1, // CHECK-NEXT: stride_h: 3, // CHECK-NEXT: filter_width: 6, // CHECK-NEXT: filter_height: 3 // CHECK-NEXT: } // CHECK-NEXT: } ] // CHECK-NEXT: name: "main" // CHECK-NEXT: } ], // CHECK-NEXT: description: "MLIR Converted.", // CHECK-NEXT: buffers: [ { // CHECK-EMPTY:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 14 16:41:28 UTC 2022 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/tests/target-annotation.mlir
} // ----- func.func @testAvgPool(%arg0: tensor<256x32x32x3xf32>) -> tensor<256x30x30x16xf32> { // CHECK: tac.device = "GPU", tac.inference_type = "FLOAT" %0 = "tfl.average_pool_2d"(%arg0) {filter_height = 3 : i32, filter_width = 3 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 1 : i32, stride_w = 1 : i32} : (tensor<256x32x32x3xf32>) -> tensor<256x30x30x16xf32> func.return %0 : tensor<256x30x30x16xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 19 19:32:06 UTC 2023 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/get-arithmetic-count.mlir
} func.func @testMaxPool2D(tensor<1x10x10x3xf32>) -> tensor<1x10x10x3xf32> { ^bb0(%arg0: tensor<1x10x10x3xf32>): // CHECK: _arithmetic_count = 2700 : i64 %0 = "tfl.max_pool_2d"(%arg0) {filter_height = 3 : i32, filter_width = 3 : i32, fused_activation_function = "RELU6", padding = "SAME", stride_h = 1 : i32, stride_w = 1 : i32} : (tensor<1x10x10x3xf32>) -> tensor<1x10x10x3xf32> func.return %0 : tensor<1x10x10x3xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 14 04:58:17 UTC 2022 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool_patterns.td
(Arith_ConstantOp ConstantAttr<RankedI32ElementsAttr<[4]>,"{0, 2, 3, 1}">)), /*filter_height*/(GetI32At<0> (GetAsVectorAttr<"kernel_size"> $attrs)), /*filter_width*/(GetI32At<1> (GetAsVectorAttr<"kernel_size"> $attrs)), /*padding*/(GetAvgPoolOpPadAttr $old_val),
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/utils/perception_ops_utils.cc
TfLitePoolParams pool_params; llvm::SmallVector<int32_t, 2> pool_size; if (failed(GetIntegerArraySafe(&func_, attrs, "pool_size", &pool_size, 2))) { return failure(); } pool_params.filter_height = pool_size[0]; pool_params.filter_width = pool_size[1]; // Retrieve strides. llvm::SmallVector<int32_t, 2> strides; if (failed(GetIntegerArraySafe(&func_, attrs, "strides", &strides, 2))) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.9K bytes - Viewed (0)