- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 34 for 16xbf16 (0.2 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
func.func private @add_scatter(%arg0: tensor<*xbf16>, %arg1: tensor<*xbf16>) -> tensor<*xbf16> { %0 = "tf.AddV2"(%arg0, %arg1) {device = ""} : (tensor<*xbf16>, tensor<*xbf16>) -> tensor<*xbf16> func.return %0 : tensor<*xbf16> } func.func private @ge_select(%arg0: tensor<*xbf16>, %arg1: tensor<*xbf16>) -> tensor<*xi1> { %0 = "tf.GreaterEqual"(%arg0, %arg1) {device = ""} : (tensor<*xbf16>, tensor<*xbf16>) -> tensor<*xi1>
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/lower_tf.mlir
func.func @SparseSoftmaxCrossEntropyWithLogits_with_dynamic(%features: tensor<*xbf16>, %labels: tensor<*xi64>) -> (tensor<2xbf16>, tensor<*xbf16>) { // CHECK: tf.SparseSoftmaxCrossEntropyWithLogits %0:2 = "tf.SparseSoftmaxCrossEntropyWithLogits"(%features, %labels) : (tensor<*xbf16>, tensor<*xi64>) -> (tensor<2xbf16>, tensor<*xbf16>) func.return %0#0, %0#1 : tensor<2xbf16>, tensor<*xbf16> } // CHECK-LABEL: func @tanhgrad_float
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 05 18:35:42 UTC 2024 - 92K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/propagate_quantize_type.mlir
} func.func private @composite_dequantize_uniform(%arg0: tensor<*xi8>) -> tensor<*xbf16> { %cst = "tf.Const"() {value = dense<1.574710e-02> : tensor<bf16>} : () -> tensor<bf16> %0 = "tf.Cast"(%arg0) {Truncate = false} : (tensor<*xi8>) -> tensor<*xbf16> %1 = "tf.Mul"(%0, %cst) : (tensor<*xbf16>, tensor<bf16>) -> tensor<*xbf16> return %1 : tensor<*xbf16> } // CHECK-LABEL: func @propagate_xladotv2_bf16
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/const-fold.mlir
%8 = "tfl.mul"(%2, %3) {fused_activation_function = "NONE"} : (tensor<4xbf16>, tensor<4xbf16>) -> tensor<4xbf16> func.return %5, %6, %7, %8 : tensor<bf16>, tensor<4xbf16>, tensor<4xbf16>, tensor<4xbf16> } // CHECK-LABEL: @mul_f16 func.func @mul_f16() -> (tensor<f16>, tensor<4xf16>, tensor<4xf16>, tensor<4xf16>) { %0 = arith.constant dense<4.5> : tensor<f16>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 45.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/convert_tensor.mlir
%2:2 = tf_executor.island wraps "tf.Const"() {device = "", dtype = bf16, value = dense<[4.900000e+01, 8.200000e+02]> : tensor<2xbf16>} : () -> tensor<bf16> loc("const3") %3:2 = tf_executor.island wraps "tf.Const"() {device = "", dtype = bf16, value = dense<0.000000e+00> : tensor<bf16>} : () -> tensor<bf16> loc("const4")
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/tensorflow/tests/mlir2graph/convert_tensor.mlir
// CHECK: dtype: DT_HALF // CHECK: half_val: 15360 // CHECK-NEXT: half_val: 16384 %2:2 = tf_executor.island wraps "tf.Const"() {device = "", dtype = bf16, value = dense<[4.900000e+01, 8.200000e+02]> : tensor<2xbf16>} : () -> tensor<bf16> loc("const3") // CHECK: name: "const3" // CHECK-NEXT: op: "Const" // CHECK: dtype: DT_BFLOAT16 // CHECK: half_val: 16964 // CHECK: half_val: 17485
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/tensorflow/cc/const_op_size_test.cc
EXPECT_THAT(num_bytes, Eq(16)); } TEST_F(GetSizeInBytesTest, Bfloat16ConstOpSizeInBytes) { constexpr absl::string_view kConstOpExpr = R"mlir( %cst = "tf.Const"() {value = dense<1.0> : tensor<7xbf16>} : () -> tensor<7xbf16> )mlir"; Block block{}; TF::ConstOp int_tensor_const_op = ParseConstOp(kConstOpExpr, block, ctx_); const int64_t num_bytes = GetSizeInBytes(int_tensor_const_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 14 04:37:13 UTC 2022 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/shape-attrs.pbtxt
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 04 18:02:53 UTC 2020 - 5K 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/python/graphdef_to_tfl_flatbuffer.cc
pass_config.legalize_custom_tensor_list_ops = toco_flags.legalize_custom_tensor_list_ops(); // Disable the unfolding of the 16x16 TF::BatchMatMulOp to avoid the // conversion to an unsupported 16x16 TFL::FullyConnectedOp. if (toco_flags.inference_type() == toco::IODataType::QUANTIZED_INT16) { pass_config.unfold_batch_matmul = false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 11 19:29:56 UTC 2024 - 5.1K bytes - Viewed (0)