Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for 2x3x3x3xf32 (0.09 sec)

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

    func.func @ConvertSliceToIdentityI32(%arg0: tensor<2x3x4x5xf32>) -> tensor<2x3x4x5xf32> {
      %begin = arith.constant dense<0> : tensor<4xi32>
      %shape = arith.constant dense<[2,3,4,5]> : tensor<4xi32>
      %0 = "tfl.slice"(%arg0, %begin, %shape) : (tensor<2x3x4x5xf32>, tensor<4xi32>, tensor<4xi32>) -> tensor<2x3x4x5xf32>
      func.return %0 : tensor<2x3x4x5xf32>
      // CHECK: return %arg0
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 20:31:41 UTC 2024
    - 284.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

    // -----
    
    // CHECK-LABEL: func @diag_part
    // CHECK-SAME: %[[ARG:.*]]: tensor<4x3x4x3xf32>
    func.func @diag_part(%arg0: tensor<4x3x4x3xf32>) -> tensor<4x3xf32> {
      // CHECK: %[[RS:.*]] = mhlo.reshape %[[ARG]] : (tensor<4x3x4x3xf32>) -> tensor<12x12xf32>
      // CHECK-DAG: %[[IOTA0:.*]] = "mhlo.iota"() <{iota_dimension = 0 : i64}> : () -> tensor<12x12xi32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
Back to top