- Sort Score
- Result 10 results
- Languages All
Results 121 - 130 of 506 for _arg0 (0.04 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/annotate-parameter-replication.mlir
// CHECK-LABEL: func @annotate_broadcast_values func.func @annotate_broadcast_values(%arg0: tensor<?xi32>) -> tensor<?xi32> { %0 = "tf._A"(%arg0) : (tensor<?xi32>) -> tensor<?xi32> %1 = "tf._B"(%arg0) : (tensor<?xi32>) -> tensor<?xi32> %5:2 = tf_device.replicate([%0, %arg0] as %ri_0: tensor<?xi32>) {n = 2 : i32} { %2 = "tf._F"(%arg0) : (tensor<?xi32>) -> tensor<?xi32> %3 = "tf.Identity"(%1) : (tensor<?xi32>) -> tensor<?xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jul 25 02:54:34 UTC 2023 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/constant-fold.mlir
} func.func @RemoveTrivialAdd(%arg0: tensor<2x2xf32>) -> tensor<2x2xf32> { %cst = arith.constant dense<0.0> : tensor<2x2xf32> %0 = "tf.Add"(%arg0, %cst) : (tensor<2x2xf32>, tensor<2x2xf32>) -> tensor<2x2xf32> func.return %0 : tensor<2x2xf32> // CHECK-LABEL: RemoveTrivialAdd // CHECK-NEXT: return %arg0 : tensor<2x2xf32> } func.func @RemoveTrivialAddBf16RHS(%arg0: tensor<2x2xbf16>) -> tensor<2x2xbf16> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 31 23:22:24 UTC 2024 - 36.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/lower_tf.mlir
%0 = "tf.Pack"(%arg0, %arg1) : (tensor<3x5xf32>, tensor<3x5xf32>) -> tensor<2x3x5xf32> func.return %0 : tensor<2x3x5xf32> } // CHECK-LABEL: func @square func.func @square(%arg0: tensor<3xf32>) -> tensor<3xf32> { // CHECK: "tf.Mul"(%arg0, %arg0) %1 = "tf.Square"(%arg0) : (tensor<3xf32>) -> tensor<3xf32> func.return %1 : tensor<3xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 05 18:35:42 UTC 2024 - 92K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize-tfl-stablehlo-max.mlir
// RUN: odml-to-stablehlo-opt %s -tfl-parse-stablehlo-ops | FileCheck %s module { func.func @main(%arg0: tensor<2xi32>) -> tensor<2xi32> attributes {tf.entry_function = {inputs = "arg0", outputs = "tfl.custom1"}} { %0 = "tfl.custom"(%arg0, %arg0) {custom_code = "stablehlo.maximum", custom_option = #tfl<const_bytes : "0x00000100002401">} : (tensor<2xi32>, tensor<2xi32>) -> tensor<2xi32> func.return %0 : tensor<2xi32> } } // CHECK: module {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 16 05:09:09 UTC 2022 - 749 bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize-tfl-stablehlo-mul.mlir
// RUN: odml-to-stablehlo-opt %s -tfl-parse-stablehlo-ops | FileCheck %s module { func.func @main(%arg0: tensor<2xi32>) -> tensor<2xi32> attributes {tf.entry_function = {inputs = "arg0", outputs = "tfl.custom1"}} { %0 = "tfl.custom"(%arg0, %arg0) {custom_code = "stablehlo.multiply", custom_option = #tfl<const_bytes : "0x00000100002401">} : (tensor<2xi32>, tensor<2xi32>) -> tensor<2xi32> func.return %0 : tensor<2xi32> } } // CHECK: module {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 16 05:09:09 UTC 2022 - 751 bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/stablehlo-custom-call-legalize-composite.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 19:49:44 UTC 2024 - 1009 bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/tests/target-annotation.mlir
} // ----- func.func @testAddReluPack(%arg0: tensor<1xf32>, %arg1: tensor<1xf32>) { // CHECK: tac.device = "GPU", tac.inference_type = "FLOAT" %0 = "tfl.add"(%arg0, %arg1) {fused_activation_function = "RELU6"} : (tensor<1xf32>, tensor<1xf32>) -> tensor<1xf32> // CHECK: tac.device = "GPU", tac.inference_type = "FLOAT"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 19 19:32:06 UTC 2023 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/batch_function_deduplicate.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/lift_quantizable_spots_as_functions_with_quantization_specs.mlir
// RUN: -split-input-file | FileCheck %s --check-prefix=STATIC-RANGE-PTQ-TO-COMPUTE-HEAVY // STATIC-RANGE-PTQ-TO-COMPUTE-HEAVY: @main func.func @main(%arg0: tensor<1x2xf32>) -> tensor<1x2xf32> { %0 = stablehlo.add %arg0, %arg0 : tensor<1x2xf32> return %0 : tensor<1x2xf32> } // Tests that `composite_add_fn_1` does not quantize when quantizing // only compute-heavy ops.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 02 18:09:38 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/lift_quantizable_spots_as_functions_drq_min_elements.mlir
func.func @lift_float_matmul(%arg0: tensor<1x12x12x512xf32>) -> (tensor<*xf32>, tensor<*xf32>) { %cst = "tf.Const"() {value = dense<0.000000e+00> : tensor<512x512xf32>} : () -> tensor<512x512xf32> %out_1 = "tf.MatMul"(%arg0, %cst) { device = "", transpose_a = false, transpose_b = false } : (tensor<1x12x12x512xf32>, tensor<512x512xf32>) -> tensor<*xf32> %out_2 = "tf.MatMul"(%arg0, %arg0) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 2.1K bytes - Viewed (0)