- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 44 for 2x2xf32 (0.15 sec)
-
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-BatchMatMulV2.mlir
// CHECK-SAME: rhs_contracting_dimensions = [1] %0 = "tf.BatchMatMulV2"(%arg0, %arg1) {adj_x = true, adj_y = true, device = ""} : (tensor<2x5xf32>, tensor<4x2xf32>) -> tensor<5x4xf32> func.return %0 : tensor<5x4xf32> } func.func @batchmatmulv2_adj_complex(%arg0: tensor<2x5xcomplex<f32>>, %arg1: tensor<4x2xcomplex<f32>>) -> tensor<5x4xcomplex<f32>> { // CHECK-LABEL: func @batchmatmulv2_adj_complex(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/tests/control_flow.mlir
} // CHECK-LABEL: pack_multiple func.func @pack_multiple(%arg0: tensor<2x3xf32>, %arg1: tensor<2x3xf32>, %arg2: tensor<2x3xf32>) -> tensor<3x2x3xf32> { %0 = "tf.MyPack"(%arg0, %arg1, %arg2) {N=3:i32, axis=0:i32} : (tensor<2x3xf32>, tensor<2x3xf32>, tensor<2x3xf32>) -> tensor<3x2x3xf32> func.return %0 : tensor<3x2x3xf32> // CHECK-NEXT: %[[AXIS:.*]] = arith.constant 0 : i32
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 10:58:25 UTC 2022 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_import_test.cc
// MLIR @main function corresponds to the TF function "main_original". OwningOpRef<ModuleOp> module_op = ParseModuleOpString(R"mlir( func.func private @main(%arg: tensor<1x2xf32>) -> (tensor<1x2xf32>) attributes {tf._original_func_name = "main_original"} { return %arg : tensor<1x2xf32> } )mlir"); ASSERT_TRUE(module_op); absl::flat_hash_map<FunctionName, FunctionAlias> function_aliases;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 07 03:47:17 UTC 2024 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/lower_quantized.mlir
// CHECK-DAG: return %[[RESULT]] func.return %0 : tensor<2x3xf32> } // CHECK-LABEL: dequantize_quint8 func.func @dequantize_quint8(%arg0: tensor<2x3x!tf_type.quint8>, %min_range: tensor<f32>, %max_range: tensor<f32>) -> tensor<2x3xf32> { // CHECK-NEXT: %[[C255:.*]] = "tf.Const"() <{value = dense<2.550000e+02> : tensor<f32>}>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize_jax_random.mlir
func.func @tfl_wrapped_jax_random_uniform(%arg0: tensor<2xui32>) -> tuple<tensor<1x2xf32>> { // This is a fake jax random uniform body. %0 = stablehlo.constant dense<0.0> : tensor<2xf32> %1 = "stablehlo.reshape"(%0) : (tensor<2xf32>) -> tensor<1x2xf32> %2 = "stablehlo.tuple"(%1) : (tensor<1x2xf32>) -> tuple<tensor<1x2xf32>> func.return %2 : tuple<tensor<1x2xf32>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/deduplicate_const.mlir
%0 = "tfl.pseudo_const" () {value = dense<[[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]]> : tensor<3x2xf32>} : () -> tensor<3x2xf32> %1 = "tfl.sub" (%0, %arg0) {fused_activation_function = "NONE"} : (tensor<3x2xf32>, tensor<3x2xf32>) -> tensor<3x2xf32> func.return %1 : tensor<3x2xf32> } } // CHECK: { // CHECK: subgraphs: [ {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 06 18:55:51 UTC 2023 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/trim-functions-tf.mlir
func.func @foo(%arg0: tensor<1x4xf32>, %arg1: tensor<1x4xf32>) -> tensor<1x4xf32> { func.return %arg0 : tensor<1x4xf32> } func.func @bar(%arg0: tensor<2x4xf32>, %arg1: tensor<2x4xf32>) -> tensor<2x4xf32> { func.return %arg0 : tensor<2x4xf32> } func.func @foobar(%arg0: tensor<1x4xf32>, %arg1: tensor<1x4xf32>) -> tensor<1x4xf32> { func.return %arg0 : tensor<1x4xf32> } // CHECK-DAG: func @main
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 26 23:53:32 UTC 2022 - 565 bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/bucketize.mlir
// CHECK-EMPTY: %0 = "tfl.pseudo_const" () {value = dense<[[-5.0, 10000.0], [150.0, 10.0], [5.0, 100.0]]> : tensor<3x2xf32>} : () -> tensor<3x2xf32> loc("Const") %1 = "tfl.bucketize"(%0) {boundaries = [0.0 : f32, 10.0 : f32, 100.0 : f32]} : (tensor<3x2xf32>) -> tensor<3x2xi32> loc("bucketize") func.return %1 : tensor<3x2xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 14 16:41:28 UTC 2022 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/post_quantize.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 18 20:32:46 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tfrt_fallback/batching_fallback.mlir
%ch0 = tfrt.new.chain %a1 = tfrt_fallback_async.const_dense_tensor dense<[[2, 2], [2, 2]]> : tensor<2x2xi32> %a2 = tfrt_fallback_async.const_dense_tensor dense<[[3, 3], [3, 3]]> : tensor<2x2xi32> %b = tfrt_fallback_async.const_dense_tensor dense<[[1, 1], [1, 1]]> : tensor<2x2xi32> // Two batch_size=2 batches get concatenated.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jul 18 22:58:56 UTC 2023 - 8.6K bytes - Viewed (0)