- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for 16x16x4xf32 (0.11 sec)
-
tensorflow/compiler/mlir/lite/tests/optimize.mlir
%1 = "tfl.arg_max"(%0, %cst) : (tensor<16x1024xf32>, tensor<1xi32>) -> tensor<16xi32> func.return %1 : tensor<16xi32> // CHECK-DAG: %[[CST:.*]] = arith.constant dense<-1> : tensor<1xi32> // CHECK: %[[SOFTMAX:.*]] = "tfl.softmax"(%arg0) <{beta = -1.000000e+00 : f32}> : (tensor<16x1024xf32>) -> tensor<16x1024xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 284.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir
// CHECK: }) : (tensor<16x1504xf32>, tensor<1xi32>, tensor<16xf32>) -> tensor<16x1504xf32> // CHECK: return %[[VAL_3]] : tensor<16x1504xf32> // CHECK: } func.func @convert_scatter_update_reshape_indices_and_updates( %arg0: tensor<16x1504xf32>, %arg1: tensor<1xi32>, %arg2: tensor<16xf32>) -> tensor<16x1504xf32> { %0 = "mhlo.scatter"(%arg0, %arg1, %arg2) ({
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 07:26:59 UTC 2024 - 340.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// ----- // CHECK-LABEL: topk_v2 // CHECK-SAME: %[[INPUT:.*]]: tensor<16x16xf32> func.func @topk_v2(%input: tensor<16x16xf32>) -> (tensor<16x8xf32>, tensor<16x8xi32>) { %k = "tf.Const"() {value = dense<8> : tensor<i32>} : () -> tensor<i32> // CHECK: chlo.top_k(%[[INPUT]], k = 8) %0:2 = "tf.TopKV2"(%input, %k): (tensor<16x16xf32>, tensor<i32>) -> (tensor<16x8xf32>, tensor<16x8xi32>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0)