- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 32 for 2xi8 (0.03 sec)
-
tensorflow/compiler/mlir/tensorflow_to_stablehlo/tests/test_tf_to_stablehlo.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 1.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize.mlir
// CHECK: %[[maxpool_i8:.*]] = "tf.MaxPool"(%[[sc1]]) // CHECK-SAME: (tensor<*xi8>) -> tensor<*xi8> // CHECK: %[[reshape_i8:.*]] = "tf.Reshape"(%[[maxpool_i8]] // CHECK-SAME: (tensor<*xi8>, tensor<2xi32>) -> tensor<*xi8> // CHECK: %[[sc2:.*]] = "quantfork.scast"(%[[reshape_i8]]) // CHECK: %[[dq:.*]] = "quantfork.dcast"(%[[sc2]]) : (tensor<*x!quant.uniform<i8:f32, 5.000000e-02:-10>>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:32:28 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_xla.mlir
func.return %11 : tensor<*xf32> } // CHECK: %[[maxpool_i8:.*]] = "tf.MaxPool" // CHECK-SAME: (tensor<*xi8>) -> tensor<*xi8> // CHECK: %[[reshape_i8:.*]] = "tf.Reshape"(%[[maxpool_i8]] // CHECK-SAME: (tensor<*xi8>, tensor<2xi32>) -> tensor<*xi8> // CHECK: %[[scast:.*]] = "quantfork.scast"(%[[reshape_i8]] // CHECK: %[[matmul:.*]] = "tf.PartitionedCall"(%[[scast]] // CHECK-SAME: f = @composite_matmul_fn_1
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:32:28 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints_test.cc
// Argument has rank 2, but its dimensions are unknown. constexpr absl::string_view kArgumentWithUnknownDims = R"mlir( func.func @unknown_dims_arg(%arg: tensor<?x?xi8>) -> tensor<?x?xi8> { return %arg : tensor<?x?xi8> } )mlir"; OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kArgumentWithUnknownDims); ASSERT_TRUE(module_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 22.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_device_ops.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 23:53:20 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/tfl_legalize_hlo.mlir
// CHECK-LABEL: func.func @convert_argmax_bool func.func @convert_argmax_bool(%arg0: tensor<2xi1>) -> tensor<i32> { %0 = "mhlo.iota"() <{iota_dimension = 0 : i64}> : () -> tensor<2xi32> %1 = mhlo.constant dense<false> : tensor<i1> %2 = mhlo.constant dense<0> : tensor<i32> %3:2 = mhlo.reduce(%arg0 init: %1), (%0 init: %2) across dimensions = [0] : (tensor<2xi1>, tensor<2xi32>, tensor<i1>, tensor<i32>) -> (tensor<i1>, tensor<i32>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 40.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/bridge/optimize.mlir
} // ----- // CHECK-LABEL: func @convolution_add_add_i8 func.func @convolution_add_add_i8( %lhs: tensor<?x3x2x1xi8>, %rhs: tensor<2x1x1x1xi8>, %zp_offset: tensor<?x2x2x1xi8>, %bias: tensor<1xi8> ) -> tensor<?x2x2x1xi8> { // CHECK-DAG: %[[conv:.*]] = mhlo.convolution // CHECK-DAG: %[[combined:.*]] = chlo.broadcast_add %[[zp_offset:.*]], %[[bias:.*]]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Feb 24 02:26:47 UTC 2024 - 10.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/ir/QuantOps.td
// operate on quantized values. // // Examples from storage to quantized type: // i8 -> !quant<"uniform[i8:f32]{1.0}"> // tensor<4xi8> -> tensor<4x!quant<"uniform[i8:f32]{1.0}">> // vector<4xi8> -> vector<4x!quant<"uniform[i8:f32]{1.0}">> def Quantization_StorageCastOp : Quantization_Op<"scast", [Pure]> { let arguments = (ins quant_RealOrStorageValueType:$arg);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 09 03:10:59 UTC 2024 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/compose-uniform-quantized-type.mlir
%11 = stablehlo.reshape %2 : (tensor<1x1x1x1xi8>) -> tensor<1xi8> %12 = stablehlo.broadcast_in_dim %11, dims = [0] : (tensor<1xi8>) -> tensor<1x3x3x4xi8> %13 = stablehlo.convert %12 : (tensor<1x3x3x4xi8>) -> tensor<1x3x3x4xf32> %14 = stablehlo.convert %3 : (tensor<3x3x4x4xi8>) -> tensor<3x3x4x4xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 37K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test.py
) ) module_str = self._extract_first_xla_call_module_op( self._output_saved_model_path ) self.assertTrue( re.search('stablehlo.dot_general.*xi8>.*xi8>.*xi32>', module_str) ) # Run quantization the first time, calibration is expected to be skipped. output_saved_model_path_2 = self.create_tempdir('output2').full_path
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 51.4K bytes - Viewed (0)