Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for kernel_size (0.6 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool.cc

      data.h_in = op_type.getShape()[2];
      data.w_in = op_type.getShape()[3];
    
      std::vector<int32_t> kernel_size;
      GetI32VectorFromDenseI64CompositeAttr(composite_attrs, "kernel_size",
                                            &kernel_size);
      data.kh = kernel_size[0];
      data.kw = kernel_size[1];
    
      std::vector<int32_t> padding;
      GetI32VectorFromDenseI64CompositeAttr(composite_attrs, "padding", &padding);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 23:16:05 UTC 2024
    - 9.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool_patterns.td

                                        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),
                                /*stride_h*/(GetI32At<0> (GetAsVectorAttr<"stride"> $attrs)),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 23:16:05 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/g3doc/space_to_depth.md

        kernel of shape [7, 7, 3, 64] to [4, 4, 12, 84].
    
        ```python
        conv0 = tf.compat.v1.layers.Conv2D(
         filters=filters,
         kernel_size=kernel_size,
         strides=2,
         padding=('SAME' if strides == 1 else 'VALID'),
         use_bias=False,
         kernel_initializer=tf.variance_scaling_initializer(),
         data_format=data_format)
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Oct 24 02:51:43 UTC 2020
    - 8.3K bytes
    - Viewed (0)
Back to top