- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for 2xui16 (0.27 sec)
-
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
} func.func @floor_mod_i16(%arg0: tensor<5xi16>, %arg1: tensor<5xi16>) -> tensor<5xi16> { %0 = "tf.FloorMod"(%arg0, %arg1) : (tensor<5xi16>, tensor<5xi16>) -> tensor<5xi16> func.return %0 : tensor<5xi16> // CHECK-LABEL: floor_mod_i16 // CHECK: "tfl.floor_mod"(%arg0, %arg1) : (tensor<5xi16>, tensor<5xi16>) -> tensor<5xi16> } func.func @exp(%arg0: tensor<5xf32>) -> tensor<5xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
let arguments = (ins TFL_TensorOf<[I8, I16, I32, I64, UI8, UI16, UI32, UI64, F32, F64]>:$input, TFL_TensorOf<[I32]>:$dilations, TFL_0DTensorOf<[I8, I16, I32, I64, UI8, UI16, UI32, UI64, F32, F64]>:$padding_value ); let results = (outs TFL_TensorOf<[I8, I16, I32, I64, UI8, UI16, UI32, UI64, F32, F64]>:$output); } def TFL_AddOp : TFL_Op<"add", [
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir
// is ui16. ui16 storage type for quantized type is not compatible with // `tfl.quantize`. func.func @uniform_quantize_op_uint16_output(%arg: tensor<2x2xf32>) -> tensor<2x2x!quant.uniform<ui16:f32, 3.000000e+0:127>> { %0 = stablehlo.uniform_quantize %arg : (tensor<2x2xf32>) -> tensor<2x2x!quant.uniform<ui16:f32, 3.000000e+0:127>> return %0 : tensor<2x2x!quant.uniform<ui16:f32, 3.000000e+0:127>> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 106.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir
// CHECK: "tf.Equal"(%arg0, %arg1) // CHECK-SAME: incompatible_shape_error = true // CHECK-SAME: device = "/job:localhost/replica:0/task:0/device:GPU:0" %0 = "tf.Equal"(%arg0, %arg1) {incompatible_shape_error = false, device = "/job:localhost/replica:0/task:0/device:GPU:0"} : (tensor<2xi32>, tensor<2xi32>) -> tensor<2xi1> func.return %0: tensor<2xi1> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 132.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/ops.mlir
func.return %0 : tensor<?xf32> } // ----- // CHECK-LABEL: testGatherNdI16Indices func.func @testGatherNdI16Indices(%arg0 : tensor<?xf32>, %arg1 : tensor<?xi16>) -> tensor<?xf32> { %0 = "tfl.gather_nd"(%arg0, %arg1) {axis = 1 : i32}: (tensor<?xf32>,tensor<?xi16>) -> tensor<?xf32> func.return %0 : tensor<?xf32> } // ----- // CHECK-LABEL: testAbs func.func @testAbs(tensor<? x f32>) -> tensor<? x f32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 189.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
ConversionPatternRewriter& rewriter) const final { RankedTensorType type = mlir::dyn_cast_or_null<RankedTensorType>(iota_op.getType()); // TF::RangeOp doesn't support UI16. if (!type || type.getElementType().isUnsignedInteger(16)) return failure(); const uint64_t dimension = iota_op.getIotaDimension(); Type element_type = type.getElementType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0)