Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for 10x500xbf16 (0.09 sec)

  1. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-with-tf2xla-hlo-importer.mlir

      }
    
      // CHECK-LABEL: approx_topk
      func.func @approx_topk(%arg0: tensor<!tf_type.resource<tensor<10x500xbf16>>> {tf._user_specified_name = "db"}) -> (tensor<10x10xbf16>) {
        %0 = "tf.ReadVariableOp"(%arg0) {device = ""} : (tensor<!tf_type.resource<tensor<10x500xbf16>>>) -> tensor<10x500xbf16>
        // CHECK: mhlo.compare  GT
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 06 15:32:52 UTC 2024
    - 38.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir

      %8 = mhlo.and %5, %7 : tensor<10x10xi1>
      %9 = mhlo.subtract %arg0, %4 : tensor<10x10xbf16>
      %10 = mhlo.divide %9, %1 : tensor<10x10xbf16>
      %11 = mhlo.add %10, %3 : tensor<10x10xbf16>
      %12 = "mhlo.select"(%8, %11, %10) : (tensor<10x10xi1>, tensor<10x10xbf16>, tensor<10x10xbf16>) -> tensor<10x10xbf16>
      %13 = "mhlo.round_nearest_afz"(%12) : (tensor<10x10xbf16>) -> tensor<10x10xbf16>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 340.2K bytes
    - Viewed (0)
Back to top