- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for PADV2 (0.18 sec)
-
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) -
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) -
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)