- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 22 for 1x2xbf16 (0.14 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/tests/cast_bf16_ops_to_f32.mlir
%cst = "tf.Const"() {device = "", value = dense<1.000000e+01> : tensor<10x2xbf16>} : () -> tensor<10x2xbf16> %0 = "tf.Cast"(%arg0) {Truncate = false, device = ""} : (tensor<1x10xf32>) -> tensor<1x10xbf16> %1 = "tf.MatMul"(%0, %cst) {device = "", transpose_a = false, transpose_b = false} : (tensor<1x10xbf16>, tensor<10x2xbf16>) -> tensor<1x2xbf16> %2 = "tf.Cast"(%1) {Truncate = false} : (tensor<1x2xbf16>) -> tensor<1x2xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/optimize_no_verify.mlir
// CHECK: return %[[V1]] : tensor<32x1x256xbf16> } // CHECK-LABEL: Relu_bf16 func.func @Relu_bf16(%arg0: tensor<2x3xbf16>) -> tensor<2x3xbf16> { %cst = arith.constant dense<0.0> : tensor<2x3xbf16> %0 = "tfl.maximum"(%arg0, %cst) : (tensor<2x3xbf16>, tensor<2x3xbf16>) -> tensor<2x3xbf16> func.return %0 : tensor<2x3xbf16> // CHECK: %[[RESULT:.*]] = "tfl.relu"(%arg0)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graph/convert_tensor.mlir
// RUN: tf-mlir-translate -mlir-to-graph %s -o - | FileCheck %s func.func @main() -> (tensor<1x2xf16>, tensor<2xf16>) { %graph:2 = tf_executor.graph { %0:2 = tf_executor.island wraps "tf.Const"() {device = "", dtype = "tfdtype$DT_HALF", value = dense<1.0> : tensor<1x2xf16>} : () -> tensor<1x2xf16> loc("const1") // CHECK: node { // CHECK-NEXT: name: "const1" // CHECK-NEXT: op: "Const" // CHECK: dtype: DT_HALF // CHECK: half_val: 15360
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 20 23:11:32 UTC 2023 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/convert_func_to_bfloat16.mlir
// CHECK-LABEL: @add_f64(%arg0: tensor<3x3xbf16>, %arg1: tensor<3x3xbf16>) -> tensor<3x3xbf16> func.func @add_f64(%arg0: tensor<3x3xf64>, %arg1: tensor<3x3xf64>) -> tensor<3x3xf64> { // CHECK-NOT: f64 // CHECK: stablehlo.add %0 = stablehlo.add %arg0, %arg1: (tensor<3x3xf64>, tensor<3x3xf64>) -> tensor<3x3xf64> return %0 : tensor<3x3xf64> } // ----- // CHECK-LABEL: @constant_f32() -> tensor<2x2xbf16>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 08 22:40:14 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/convert_tensor.mlir
// RUN: tf-mlir-translate -mlir-to-graphdef %s -o - | FileCheck %s func.func @main() -> (tensor<1x2xf16>, tensor<2xf16>) { %graph:2 = tf_executor.graph { %0:2 = tf_executor.island wraps "tf.Const"() {device = "", dtype = "tfdtype$DT_HALF", value = dense<1.0> : tensor<1x2xf16>} : () -> tensor<1x2xf16> loc("const1")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 12:28:56 UTC 2022 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/quantize-dynamic-range-float16.mlir
// CHECK: %[[DQ_2:.*]] = "tfl.dequantize"({{.*}}) : (tensor<1x1xf16>) -> tensor<1x1xf32> // CHECK: %[[DQ_3:.*]] = "tfl.dequantize"({{.*}}) : (tensor<1x1xf16>) -> tensor<1x1xf32> // CHECK: %[[DQ_4:.*]] = "tfl.dequantize"({{.*}}) : (tensor<1x1xf16>) -> tensor<1x1xf32> // CHECK: %[[DQ_5:.*]] = "tfl.dequantize"({{.*}}) : (tensor<1x1xf16>) -> tensor<1x1xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/ifrt/sink_variable_as_named_array.mlir
// module { func.func @serving_default() -> tensor<1x3xbf16> { %cst = "tf.Const"() <{value = dense<[1]> : tensor<1xi32>}> : () -> tensor<1xi32> %0 = "tf.VarHandleOp"() <{container = "", shared_name = "Variable"}> : () -> tensor<!tf_type.resource<tensor<2x3xbf16>>> %1 = "tf.ResourceGather"(%0, %cst) <{batch_dims = 0 : i64, validate_indices = true}> : (tensor<!tf_type.resource<tensor<2x3xbf16>>>, tensor<1xi32>) -> tensor<1x3xbf16>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 15:33:17 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/constant-fold.mlir
} func.func @RemoveTrivialAddBf16RHS(%arg0: tensor<2x2xbf16>) -> tensor<2x2xbf16> { %cst = arith.constant dense<0.0> : tensor<2x2xbf16> %0 = "tf.Add"(%arg0, %cst) : (tensor<2x2xbf16>, tensor<2x2xbf16>) -> tensor<2x2xbf16> func.return %0 : tensor<2x2xbf16> // CHECK-LABEL: RemoveTrivialAdd // CHECK-NEXT: return %arg0 : tensor<2x2xbf16> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 31 23:22:24 UTC 2024 - 36.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration_test.cc
TEST_F(MlirTranslationTest, TranslatesMlirToGraph) { static constexpr char kMlirSource[] = R"( func.func @main() -> (tensor<1x2xf16>, tensor<2xf16>) { %graph:2 = tf_executor.graph { %0:2 = tf_executor.island wraps "tf.Const"() {device = "", dtype = "tfdtype$DT_HALF", value = dense<1.0> : tensor<1x2xf16>} : () -> tensor<1x2xf16> loc("const1")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 20 23:11:32 UTC 2023 - 3.2K 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)