Search Options

Results per page
Sort
Preferred Languages
Advance

Results 61 - 65 of 65 for 2x4xi32 (0.23 sec)

  1. tensorflow/compiler/mlir/lite/tests/quantize.mlir

    func.func @NotQuantizeFloatConst() -> tensor<2x2xf32> {
      %0 = arith.constant dense<-0.1> : tensor<2x2xf32>
      %1 = "tfl.quantize"(%0) {qtype = tensor<2x2x!quant.uniform<u8:f32, 7.8431372549019615E-4:128>>} : (tensor<2x2xf32>) -> tensor<2x2x!quant.uniform<u8:f32, 7.8431372549019615E-4:128>>
      %2 = "tfl.dequantize"(%1) : (tensor<2x2x!quant.uniform<u8:f32, 7.8431372549019615E-4:128>>) -> tensor<2x2xf32>
      func.return %2 : tensor<2x2xf32>
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 23:10:13 UTC 2024
    - 39.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir

    func.func @dynamic_slice_float(%arg0: tensor<4x4xf32>, %arg1: tensor<i64>, %arg2: tensor<i64>) -> tensor<2x1xf32> {
      %0 = "stablehlo.dynamic_slice"(%arg0, %arg1, %arg2) {
        slice_sizes = array<i64: 2, 1>
      } : (tensor<4x4xf32>, tensor<i64>, tensor<i64>) -> tensor<2x1xf32>
      return %0 : tensor<2x1xf32>
    }
    
    // CHECK-LABEL: func @dynamic_slice_float
    // CHECK: stablehlo.dynamic_slice
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 17:10:32 UTC 2024
    - 106.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc

    //     (tensor<i32>, tensor<i32>, tensor<i32>) -> tensor<5xi32>
    //   %shape = "tf.Const"() {value = dense<[5, 1]> : tensor<2xi32>}
    //   %indices = "tf.Reshape"(%x, %shape) : (tensor<5xi32, tensor<2xi32) ->
    //     tensor<5x1xi32>
    //   "tf.TensorScatterUpdate"(%x, %indices, %updates) :
    //     (tensor<5xi32>, tensor<5x1xi32>, tensor<5xi32>) -> tensor<5xi32>
    //
    class LowerInvertPermutationOp : public RewritePattern {
     public:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 74.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/quantize_composite_functions.mlir

    // CHECK: %[[CONST_0:.+]] = stablehlo.constant dense<3.000000e-01> : tensor<2x3xf32>
    // CHECK: %[[CALL:.+]] = call @composite_dot_general_fn(%[[ARG_0]], %[[CONST_0]]) : (tensor<1x2xf32>, tensor<2x3xf32>) -> tensor<1x3xf32>
    // CHECK: return %[[CALL]]
    
      func.func private @composite_dot_general_fn(%arg0: tensor<1x2xf32>, %arg1: tensor<2x3xf32>) -> tensor<1x3xf32> attributes {_from_xla_call_module} {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 05:56:10 UTC 2024
    - 91.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

    }
    
    // Returns a RankedTensorType which is similar to `input_type` but replaces the
    // dimension size of `dim` with `dim_size`.  For example,
    // `SubstituteRankedTensorTypeDimSize(tensor<3x4xi32>, 1, 2)` returns
    // `tensor<3x2xi32>`.
    static RankedTensorType SubstituteRankedTensorTypeDimSize(
        RankedTensorType input_type, int64_t dim, int64_t dim_size) {
      auto shape = input_type.getShape().vec();
      shape[dim] = dim_size;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
Back to top