Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for PADV2 (0.04 sec)

  1. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

      // CHECK-SAME: edge_padding_high = dense<2> : tensor<1xi64>,
      // CHECK-SAME: edge_padding_low = dense<1> : tensor<1xi64>,
      // CHECK-SAME: interior_padding = dense<0> : tensor<1xi64>
      %1 = "tf.PadV2"(%arg0, %padding, %arg1) : (tensor<3xf32>, tensor<1x2xi64>, tensor<f32>) -> tensor<6xf32>
      func.return %1 : tensor<6xf32>
    }
    
    // -----
    
    // CHECK-LABEL: func @padv2_2D
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir

    // CHECK-SAME:                      %[[VAL_1:.*]]: tensor<f32>) -> tensor<11x131xf32> {
    // CHECK:           %[[VAL_2:.*]] = arith.constant dense<{{\[\[}}1, 2], [0, 3]]> : tensor<2x2xi64>
    // CHECK:           %[[VAL_3:.*]] = "tf.PadV2"(%[[VAL_0]], %[[VAL_2]], %[[VAL_1]]) : (tensor<8x128xf32>, tensor<2x2xi64>, tensor<f32>) -> tensor<11x131xf32>
    // CHECK:           return %[[VAL_3]] : tensor<11x131xf32>
    // CHECK:         }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 340.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

      return DenseIntElementsAttr::get(
          tensorflow::GetTypeFromTFTensorShape({shape[0]}, element_type), values);
    }
    
    // Returns interior padding to use in HLO Pad op based on the TensorFlow padding
    // in TensorFlow PadV2 op.
    static DenseIntElementsAttr GetInteriorPadding(ElementsAttr tf_padding) {
      auto length = tf_padding.getShapedType().getShape()[0];
      auto element_type = IntegerType::get(tf_padding.getContext(), 64);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
Back to top