- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 190 for 20xi32 (0.11 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/xla_call_module_serialization.mlir
return %arg2 : tensor<?xi32> } // CHECK-LABEL: func @main // CHECK-SAME: %[[ARG0:.*]]: tensor<10xi32>, %[[ARG1:.*]]: tensor<10xi32> func.func @main(%arg0: tensor<10xi32>, %arg1: tensor<10xi32>) -> tensor<10xi32> { // CHECK: %[[RESULT:.*]] = "tf.XlaCallModule"(%[[ARG0]], %[[ARG1]]) // CHECK-SAME: Sout = [#tf_type.shape<?>] // CHECK-SAME: dim_args_spec = [] // CHECK-NOT: _entry_function
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 26 03:48:35 UTC 2023 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/xla_call_module_deserialization.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 02 18:38:51 UTC 2023 - 6.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/xla_call_module_round_trip.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 02 18:38:51 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize-tfl-stablehlo-sub.mlir
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.subtract", 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/tf-tfl-translate-serialize-stablehlo-division.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 24 20:01:54 UTC 2023 - 306 bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/tf_add.mlir
func.func @main(%arg0: tensor<10xi32>, %arg1: tensor<10xi32>) -> tensor<10xi32> attributes {tf.entry_function = {inputs = "input0,input1", outputs = "Add"}} { %graph = tf_executor.graph { %2:2 = tf_executor.island wraps "tf.Add"(%arg0, %arg1) {T = "tfdtype$DT_INT32", device = ""} : (tensor<10xi32>, tensor<10xi32>) -> tensor<10xi32> loc("Add") tf_executor.fetch %2 : tensor<10xi32> } func.return %graph : tensor<10xi32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 22:54:55 UTC 2023 - 1.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/inlining.mlir
// Inline a function that contains only tfl ops. func.func @func_with_tfl_ops(%arg0 : tensor<2xi32>) -> tensor<2xi32> { %0 = "tfl.sub"(%arg0, %arg0) {fused_activation_function = "RELU6"} : (tensor<2xi32>, tensor<2xi32>) -> tensor<2xi32> %1 = "tfl.add"(%0, %arg0) {fused_activation_function = "RELU6"} : (tensor<2xi32>, tensor<2xi32>) -> tensor<2xi32> func.return %1: tensor<2xi32> } // CHECK-LABEL: func @inline_with_arg
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 14:24:59 UTC 2022 - 1000 bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/extract_outside_compilation.mlir
%5:2 = "tf.C"(%4) : (tensor<2xi32>) -> (tensor<2xi32>, tensor<2xi32>) tf_device.return %5#0, %5#1 : tensor<2xi32>, tensor<2xi32> }) {num_cores_per_replica = 1, topology = "", device_assignment = []} : () -> (tensor<2xi32>, tensor<2xi32>) tf_device.return %2#1 : tensor<2xi32> } func.return %1 : tensor<2xi32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 129.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/inlining.mlir
func.func private @simple_callee() -> tensor<2xi32> { %cst = "tf.Const"() { value = dense<2> : tensor<2xi32> } : () -> tensor<2xi32> func.return %cst : tensor<2xi32> } func.func private @simple_callee_with_noinline() -> tensor<2xi32> attributes {tf._noinline = true} { %cst = "tf.Const"() { value = dense<2> : tensor<2xi32> } : () -> tensor<2xi32> func.return %cst : tensor<2xi32> } // Test that simple TF operations can be inlined.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/lower_tf.mlir
// CHECK-DAG: %[[INDICES_3:.*]] = "tf.Mul"(%[[INDICES_2]], %arg0) : (tensor<5xi32>, tensor<5xi32>) -> tensor<5xi32> // CHECK-DAG: %[[INDICES_4:.*]] = "tf.TensorScatterAdd"(%[[cst_3]], %0, %[[UPDATES]]) : (tensor<5xi32>, tensor<5x1xi32>, tensor<5xi32>) -> tensor<5xi32> // CHECK-DAG: %[[INDICES_5:.*]] = "tf.AddV2"(%[[INDICES_3]], %[[INDICES_4]]) : (tensor<5xi32>, tensor<5xi32>) -> tensor<5xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 05 18:35:42 UTC 2024 - 92K bytes - Viewed (0)