Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for 5x7xi1 (0.13 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/lower_tf.mlir

      func.return %0 : tensor<3x4xi1>
    }
    
    // CHECK-LABEL: @is_nan
    func.func @is_nan(%arg0: tensor<3x4xf32>) -> tensor<3x4xi1> {
      // CHECK: %[[RESULT:.*]] = "tf.NotEqual"(%arg0, %arg0) <{incompatible_shape_error = true}> : (tensor<3x4xf32>, tensor<3x4xf32>) -> tensor<3x4xi1>
      %0 = "tf.IsNan"(%arg0) : (tensor<3x4xf32>) -> tensor<3x4xi1>
      // CHECK: return %[[RESULT]]
      func.return %0 : tensor<3x4xi1>
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jan 05 18:35:42 UTC 2024
    - 92K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/unroll-batch-matmul.mlir

    // -----
    
    func.func @batchMatMulV3MatrixInt8(%arg0: tensor<4x5xi8>, %arg1: tensor<5x6xi8>) -> tensor<4x6xi32> {
      %0 = "tf.BatchMatMulV3"(%arg0, %arg1) : (tensor<4x5xi8>, tensor<5x6xi8>) -> tensor<4x6xi32>
      func.return %0 : tensor<4x6xi32>
    
      // CHECK-LABEL: batchMatMulV3MatrixInt8
      // CHECK: %0 = "tf.BatchMatMulV3"(%arg0, %arg1) : (tensor<4x5xi8>, tensor<5x6xi8>) -> tensor<4x6xi32>
      // CHECK: return %0 : tensor<4x6xi32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Dec 06 18:42:28 UTC 2023
    - 63.7K bytes
    - Viewed (0)
Back to top