- Sort Score
- Result 10 results
- Languages All
Results 11 - 16 of 16 for 8x16xf32 (0.25 sec)
-
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/fallback.mlir
%1 = "tf.MatMul"(%arg0, %0) {T = f32, device = "/device:CPU:0", transpose_a = false, transpose_b = false} : (tensor<3x1xf32>, tensor<1x3xf32>) -> tensor<3x3xf32> func.return %1 : tensor<3x3xf32> } // CHECK-LABEL: func @gpu_device func.func @gpu_device(%arg0: tensor<3x1xf32>, %arg1: tensor<!tf_type.resource<tensor<1x3xf32>>>) -> tensor<3x3xf32> { // CHECK: {{%.*}} = corert.get_op_handler %arg0 "/device:GPU:0"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/device_conversion.mlir
// RUN: tf-tfrt-opt -tf-to-tfrt %s | FileCheck %s --dump-input=fail // CHECK-LABEL: func @device_test func.func @device_test( %arg0: tensor<3x1xf32> {tf_saved_model.index_path = [0]}, %arg1: tensor<1x3xf32> {tf_saved_model.index_path = [0]}) -> (tensor<3x3xf32> {tf_saved_model.index_path = []}) { // CHECK: {{%.*}} = corert.get_op_handler %arg0 "/device:GPU:0"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 645 bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/shared_variable_v1.py
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:49:35 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/basic.mlir
// CHECK-SAME: [[arg0:%.*]]: !tfrt_fallback.tf_tensor, // CHECK-SAME: [[arg1:%.*]]: !tfrt_fallback.tf_tensor) // CHECK-SAME: -> (!tfrt.chain, !tfrt_fallback.tf_tensor) func.func @func_basic( %arg0: tensor<3x1xf32> {tf_saved_model.index_path = [0]}, %arg1: tensor<!tf_type.resource<tensor<1x3xf32>>> {tf_saved_model.index_path = [1]}) -> (tensor<3x3xf32> {tf_saved_model.index_path = []})
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/cast_bf16_ops_to_f32.mlir
// CHECK: return %[[identity]] : tensor<1x3x2x2xf32> func.func @cast_bf16_matmul_to_fp32(%arg0: tensor<1x10xf32>) -> (tensor<1x2xf32>) { %cst = "tf.Const"() {device = "", value = dense<1.000000e+01> : tensor<10x2xbf16>} : () -> tensor<10x2xbf16> %0 = "tf.Cast"(%arg0) {Truncate = false, device = ""} : (tensor<1x10xf32>) -> tensor<1x10xbf16>
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/experimental/tac/tests/target-annotation.mlir
%1 = "tfl.pseudo_const"() {value = dense<1.000000e+00> : tensor<16xf32>} : () -> tensor<16xf32> // CHECK: tac.device = "GPU", tac.inference_type = "FLOAT" %2 = "tfl.conv_2d"(%arg0, %0, %1) {dilation_h_factor = 1 : i32, dilation_w_factor = 1 : i32, fused_activation_function = "NONE", padding = "VALID", stride_h = 1 : i32, stride_w = 1 : i32} : (tensor<256x32x32x3xf32>, tensor<16x3x3x3xf32>, tensor<16xf32>) -> tensor<256x30x30x16xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 19 19:32:06 UTC 2023 - 6.2K bytes - Viewed (0)