Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for 1x5x5x2xf32 (0.1 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir

    // CHECK-LABEL: func @tfAssertTrue
    func.func @tfAssertTrue(%arg0: tensor<1x1x6x2xf32>) {
      %t = arith.constant dense<true> : tensor<i1>
      // CHECK-NOT: tf.Assert
      "tf.Assert"(%t, %arg0) {summarize = 3} : (tensor<i1>, tensor<1x1x6x2xf32>) -> ()
      func.return
    }
    
    // CHECK-LABEL: func @tfAssertFalse
    func.func @tfAssertFalse(%arg0: tensor<1x1x6x2xf32>) {
      %f = arith.constant dense<false> : tensor<i1>
      // CHECK: tf.Assert
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 132.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir

    }
    
    func.func @gatherV2VectorIndices_I64Axis(%arg0 : tensor<1x2x20xf32>, %arg1 : tensor<3x5xi32>) -> tensor<1x3x5x20xf32> {
      %0 = "tf.Const"() { value = dense<[1]> : tensor<1xi64> } : () -> tensor<1xi64>
      %1 = "tf.GatherV2"(%arg0, %arg1, %0) : (tensor<1x2x20xf32>, tensor<3x5xi32>, tensor<1xi64>) -> tensor<1x3x5x20xf32>
      func.return %1 : tensor<1x3x5x20xf32>
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 05 01:54:33 UTC 2024
    - 153.4K bytes
    - Viewed (0)
Back to top