- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for 1x2x2xf32 (0.18 sec)
-
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
func.return %0 : tensor<2x2xf32> // CHECK-LABEL:squeezeDefault // CHECK: "tfl.squeeze"(%arg0) <{squeeze_dims = []}> : (tensor<1x2x2xf32>) -> tensor<2x2xf32> } func.func @squeezeSingleAxis(%arg0: tensor<2x1x2xf32>) -> tensor<2x2xf32> { %0 = "tf.Squeeze"(%arg0) {squeeze_dims = [1]} : (tensor<2x1x2xf32>) -> tensor<2x2xf32> func.return %0 : tensor<2x2xf32> // CHECK-LABEL:squeezeSingleAxis
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
// CHECK-SAME: ({{%.+}}: tensor<1x2x3xf32>) // CHECK-SAME: -> (tensor<1x8x3xf32>, tensor<1x8x3xf32>) func.func @while_shape_invariant_different_dims(%arg0: tensor<1x2x3xf32>) -> (tensor<1x8x3xf32>, tensor<1x8x3xf32>) { // CHECK: "tf.While" // CHECK-SAME: (tensor<1x2x3xf32>) // CHECK-SAME: -> tensor<1x8x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir
%1 = "tf.TensorListGetItem"(%0, %arg2, %arg1) {device = "/job:localhost/replica:0/task:0/device:GPU:0"} : (tensor<!tf_type.variant<tensor<1x32xf32>>>, tensor<i32>, tensor<2xi32>) -> tensor<1x32xf32> %2 = "tf.TensorListGetItem"(%0, %arg3, %arg1) {device = "/job:localhost/replica:0/task:0/device:GPU:0"} : (tensor<!tf_type.variant<tensor<1x32xf32>>>, tensor<i32>, tensor<2xi32>) -> tensor<1x32xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 132.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
// ----- func.func @testDynamicStitch(%arg0: tensor<2x2xf32>) -> tensor<2x2xf32> { %indices = "tf.Const"() {value = dense<[1, 0]> : tensor<2xi32>} : () -> tensor<2xi32> %0 = "tf.DynamicStitch"(%indices, %arg0) : (tensor<2xi32>, tensor<2x2xf32>) -> tensor<2x2xf32> func.return %0 : tensor<2x2xf32> } // ----- func.func @testDynamicStitch() -> tensor<2x2xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
return success(); } }; // Fuses Unpack with proceeding Concatenation to Reshape if output type has // static shape and activation function is none. For example: // // // %input: tensor<1x3x2xf32> // %unpack:3 = "tfl.unpack"(%input) {axis = 1 : i32, num = 3 : i32} // %res = "tfl.concatenation"(%unpack#0, %unpack#1, %unpack#2) // {axis = -1 : i32, fused_activation_function = "NONE"} //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0)