- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 100 for 5x3xi32 (0.55 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/cc/pre_calibration_test.cc
module attributes {} { func.func @main(%arg0: tensor<1x4xf32>) -> tensor<1x3xf32> attributes {} { %0 = stablehlo.constant dense<1.0> : tensor<4x3xf32> %1 = stablehlo.dot_general %arg0, %0, contracting_dims = [1] x [0], precision = [DEFAULT, DEFAULT] : (tensor<1x4xf32>, tensor<4x3xf32>) -> tensor<1x3xf32> return %1 : tensor<1x3xf32> } } )mlir"); ASSERT_TRUE(module_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 21:41:08 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-binary-elementwise.mlir
// fixed upstream). func.func @broadcast_add(%arg0: tensor<1xi32>, %arg1: tensor<1x2xi32>) -> tensor<1x2xi32> { // CHECK-NEXT: %[[LHS_BCAST:.+]] = "mhlo.broadcast_in_dim"(%arg0) <{broadcast_dimensions = dense<1> : tensor<1xi64>}> // CHECK-NEXT: mhlo.add %[[LHS_BCAST]], %arg1 %0 = "tf.AddV2"(%arg0, %arg1) : (tensor<1xi32>, tensor<1x2xi32>) -> tensor<1x2xi32> func.return %0: tensor<1x2xi32> } // CHECK-LABEL: func @broadcast_multi_dim_add
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 18.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu-resource-read-for-write.mlir
func.return } func.func private @write_chain_func(%arg0: tensor<1xf32>) -> (tensor<1xf32>) { %cst = "tf.Const"() <{value = dense<[[0, 1]]> : tensor<1x2xi32>}> : () -> tensor<1x2xi32> %0 = "tf.XlaAllReduce"(%arg0, %cst) <{mode = "CrossReplica", reduce_op = "Add"}> : (tensor<1xf32>, tensor<1x2xi32>) -> tensor<1xf32> return %0 : tensor<1xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 16:54:40 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/layout_optimization_move_transposes_begin.mlir
%2 = "tf.Transpose"(%1, %cst_1) {device = ""} : (tensor<8x64x64xf32>, tensor<3xi32>) -> tensor<64x8x64xf32> %3 = "tf.Reshape"(%2, %cst_2) {device = ""} : (tensor<64x8x64xf32>, tensor<2xi32>) -> tensor<512x64xf32> func.return %3 : tensor<512x64xf32> // CHECK-DAG: %[[CST_0:.*]] = "tf.Const"() <{value = dense<[2, 0, 1]> : tensor<3xi32>}> : () -> tensor<3xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-include-tf2xla-fallback.mlir
%0 = mhlo.constant dense<[[1, 1], [2, 2]]> : tensor<2x2xi32> // NO_FALLBACK: tf.MirrorPad // SUPPORTED_FALLBACK_DEVICE-NOT: tf.MirrorPad // UNSPECIFIED_FALLBACK_DEVICE: tf.MirrorPad // UNSUPPORTED_FALLBACK_DEVICE: tf.MirrorPad %1 = "tf.MirrorPad"(%arg0, %0) {mode = "SYMMETRIC"} : (tensor<2x3xcomplex<f64>>, tensor<2x2xi32>) -> tensor<4x7xcomplex<f64>> func.return %1 : tensor<4x7xcomplex<f64>> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 16 19:04:03 UTC 2023 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/const-values.pbtxt
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 24 00:20:25 UTC 2020 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/prepare-composite-functions-tf.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 122.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/functional-control-flow-to-cfg.mlir
// CHECK-LABEL: func @testWhileCasts(%arg0: tensor<!tf_type.variant<tensor<1x3xf32>>>) -> tensor<!tf_type.variant<tensor<*xf32>>> func.func @testWhileCasts(%arg0: tensor<!tf_type.variant<tensor<1x3xf32>>>) -> (tensor<!tf_type.variant<tensor<*xf32>>>) { %0 = "tf.While"(%arg0) { cond = @testWhileCond, body = @testWhileBody, is_stateless = false } : (tensor<!tf_type.variant<tensor<1x3xf32>>>) -> (tensor<!tf_type.variant<tensor<*xf32>>>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call_test.cc
%0 = stablehlo.dot_general %arg0, %arg1, contracting_dims = [1] x [0], precision = [DEFAULT, DEFAULT] : (tensor<1x1024xf32>, tensor<1024x3xf32>) -> tensor<1x3xf32> return %0 : tensor<1x3xf32> } } )mlir"; TEST_F(LiftAsFunctionCallTest, LiftedFunctionSucceeds) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 26.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/xla_broadcast.mlir
// CHECK-NEXT: %[[GROUP:.*]] = "tf.Const"() // CHECK-SAME: [0, 1, 2, 3] // CHECK-NEXT: %[[REDUCED:.*]] = "tf.XlaAllReduce"(%[[ID]], %[[GROUP]]) <{mode = "CrossReplica", reduce_op = "Add"}> : (tensor<f32>, tensor<1x4xi32>) -> tensor<f32> // CHECK-NEXT: "tf.OpA"(%[[REDUCED]]) : (tensor<f32>) -> () tf_device.replicate {n = 4 : i32} { "tf_device.cluster"() ({ "tf.OpA"(%arg0) : (tensor<f32>) -> () tf_device.return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 2.9K bytes - Viewed (0)