- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for 8x10xf32 (0.19 sec)
-
tensorflow/compiler/mlir/lite/tests/optimize.mlir
%2 = "tfl.exp"(%1) : (tensor<8x128xf32>) -> tensor<8x128xf32> %3 = "tfl.sum"(%2, %cst) {keep_dims = true} : (tensor<8x128xf32>, tensor<1xi32>) -> tensor<8x1xf32> %4 = "tfl.div"(%2, %3) {fused_activation_function = "NONE"} : (tensor<8x128xf32>, tensor<8x1xf32>) -> tensor<8x128xf32> func.return %4 : tensor<8x128xf32> // CHECK-LABEL: SoftMaxWithNormalization
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/tf2xla/tests/legalize-tf.mlir
// CHECK-NEXT: return %[[DIV]] : tensor<4x10xf32> %0 = "tf.Softsign"(%arg0) : (tensor<4x10xf32>) -> tensor<4x10xf32> func.return %0 : tensor<4x10xf32> } // ----- // CHECK-LABEL: func @softsign_grad func.func @softsign_grad(%arg0: tensor<4x10xf32>, %arg1: tensor<4x10xf32>) -> tensor<4x10xf32> { // CHECK-NEXT: %[[ONE:.*]] = mhlo.constant dense<1.000000e+00> : tensor<f32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir
func.func @torch_index_select(%arg0: tensor<2x1xf32>, %arg1: tensor<2xi32>) -> tensor<2x1xf32> { %0 = "mhlo.torch_index_select"(%arg0, %arg1) { batch_dims = 0 : i64, dim = 0 : i64 } : (tensor<2x1xf32>, tensor<2xi32>) -> tensor<2x1xf32> func.return %0 : tensor<2x1xf32> } // CHECK-LABEL: func @lowered_cumsum( // CHECK-SAME: %[[VAL_0:.*]]: tensor<4x12xf32>) -> tensor<4x12xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 07:26:59 UTC 2024 - 340.2K bytes - Viewed (0)