- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for 4x4xf32 (0.24 sec)
-
tensorflow/compiler/mlir/lite/tests/split-merged-operands.mlir
func.return %2 : tensor<4x4x4xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/components/pre_calibration_component.mlir
func.func @main(%arg0: tensor<1x4xf32>) -> tensor<1x3xf32> { %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> } // CHECK: @main(%[[ARG_0:.+]]: tensor<1x4xf32>) -> tensor<1x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/quantize/quantize.mlir
// CHECK: return func.func private @composite_dot_general_fn(%arg0: tensor<1x4xf32>, %arg1: tensor<4x3xf32>) -> tensor<1x3xf32> attributes {_from_xla_call_module} { %0 = stablehlo.dot_general %arg0, %arg1, contracting_dims = [1] x [0] : (tensor<1x4xf32>, tensor<4x3xf32>) -> tensor<1x3xf32> return %0 : tensor<1x3xf32> } } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 18 01:38:40 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/restore_function_name.mlir
// CHECK-SAME: %[[ARG3:[^:[:space:]]+]] func.func private @main(%arg0: tensor<1x4xf32>, %arg1: tensor<4x3xf32>) -> tensor<1x3xf32> attributes {_from_xla_call_module} { %0 = stablehlo.dot_general %arg0, %arg1, contracting_dims = [1] x [0] : (tensor<1x4xf32>, tensor<4x3xf32>) -> tensor<1x3xf32> return %0 : tensor<1x3xf32> // CHECK: %[[DOT:.+]] = stablehlo.dot_general %[[ARG2]], %[[ARG3]] // CHECK: return %[[DOT]] }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 08 22:40:14 UTC 2024 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-BatchMatMulV2.mlir
%0 = "tf.BatchMatMulV2"(%arg0, %arg1) {T = f32, adj_x = false, adj_y = false, device = ""} : (tensor<3x4x2xf32>, tensor<2x4xf32>) -> tensor<3x4x4xf32> func.return %0 : tensor<3x4x4xf32> } func.func @batchmatmulv2_rhs_batch(%arg0: tensor<4x2xf32>, %arg1: tensor<3x2x4xf32>) -> tensor<3x4x4xf32> { // CHECK-LABEL: func @batchmatmulv2_rhs_batch
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/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/lite/tests/legalize_jax_random.mlir
func.func @tfl_wrapped_jax_random_normal(%arg0: tensor<2xui32>) -> tuple<tensor<3x4xf32>> { // This is a fake jax random normal body. %0 = stablehlo.constant dense<0.0> : tensor<12xf32> %1 = "stablehlo.reshape"(%0) : (tensor<12xf32>) -> tensor<3x4xf32> %2 = "stablehlo.tuple"(%1) : (tensor<3x4xf32>) -> tuple<tensor<3x4xf32>> func.return %2 : tuple<tensor<3x4xf32>> } // CHECK-LABEL: func @tfl_wrapped_jax_random_uniform(
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/stablehlo/tests/tfl_legalize_hlo_custom_call.mlir
%0 = mhlo.custom_call @tf.ResizeBilinear(%arg0) { backend_config = "this_is_a_test_string" } : (tensor<1x4xf32>) -> (tensor<1x8xf32>) func.return %0 : tensor<1x8xf32> // CHECK: %0 = mhlo.custom_call @tf.ResizeBilinear(%arg0) { // CHECK-SAME: backend_config = "this_is_a_test_string" // CHECK-SAME: } : (tensor<1x4xf32>) -> tensor<1x8xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/cast_bf16.mlir
func.func @main(tensor<4x5xbf16>) -> tensor<4x5xbf16> { ^bb0(%arg0: tensor<4x5xbf16>): // CHECK-LABEL: @main // CHECK: (tensor<4x5xbf16>) -> tensor<4x5xf32> // CHECK-NEXT: (tensor<4x5xf32>) -> tensor<4x5xbf16> %0 = "tfl.cast" (%arg0) : (tensor<4x5xbf16>) -> tensor<4x5xf32> loc("cast1") %1 = "tfl.cast" (%0) : (tensor<4x5xf32>) -> tensor<4x5xbf16> loc("cast2") func.return %1 : tensor<4x5xbf16>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 18 21:28:19 UTC 2024 - 596 bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/quantize-dynamic-range-float16.mlir
time_major = false} : ( tensor<1x2x3xf32>, tensor<1x1xf32>, tensor<1x1xf32>, tensor<1x1xf32>, tensor<1x1xf32>, tensor<1x1xf32>, tensor<1x1xf32>, tensor<1x1xf32>, tensor<1x1xf32>, none, none, none, tensor<3xf32>, tensor<3xf32>, tensor<3xf32>, tensor<3xf32>, none, none, tensor<1x3xf32>, tensor<1x3xf32>, none, none, none, none) -> tensor<1x2x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 4.6K bytes - Viewed (0)