- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 193 for _arg0 (0.03 sec)
-
tensorflow/compiler/mlir/tfrt/tests/ifrt/rewrite_cluster_to_ifrt_call.mlir
return %0 : tensor<1x3xf32> } // CHECK-LABEL: @_func func.func private @_func(%arg0: tensor<1x3xf32>) -> (tensor<1x3xf32>) { return %arg0 : tensor<1x3xf32> } } // ----- // SPMD: one input and no return // // CHECK-LABEL: func.func @serving_default(%arg0: tensor<1x3xf32>) { // CHECK-NEXT: "tf.IfrtCall"(%arg0) // CHECK-SAME: {program_id = [[PROGRAM_ID:.*]] : i64, variable_arg_indices = []}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Feb 17 07:28:40 UTC 2024 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir
// CHECK: return %[[VAL_3]] : tensor<2xf32> // CHECK: } func.func @floordiv_f32(%arg0: tensor<2xf32>) -> tensor<2xf32> { %0 = mhlo.divide %arg0, %arg0 : tensor<2xf32> %1 = mhlo.divide %arg0, %arg0 : tensor<2xf32> %2 = "mhlo.floor"(%1) : (tensor<2xf32>) -> tensor<2xf32> func.return %2 : tensor<2xf32> } // CHECK-LABEL: func @floordiv_f16_broadcast(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 07:26:59 UTC 2024 - 340.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/reconfig_batch_op.mlir
// CHECK-LABEL: func private @batched_function func.func private @batched_function(%arg0: tensor<1x3xf32>) -> tensor<1x3xf32> { %2 = "tf.Identity"(%arg0) : (tensor<1x3xf32>) -> tensor<1x3xf32> func.return %2 : tensor<1x3xf32> } // CHECK-LABEL: func @main func.func @main(%arg0: tensor<1x3xf32>) -> tensor<*xf32> { // CHECK: "tf.BatchFunction" // CHECK-SAME: allowed_batch_sizes = [6]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 17:38:34 UTC 2024 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/nchw_convolution_to_nhwc.mlir
// CHECK-LABEL: nchw_conv // CHECK-SAME: %[[ARG:.+]]: tensor<1x8x4x4xf32> func.func @nchw_conv(%arg0: tensor<1x8x4x4xf32>) -> tensor<1x8x4x4xf32> { %0 = stablehlo.constant() {value = dense<7.000000e+00> : tensor<8x8x3x3xf32>} : () -> tensor<8x8x3x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 23:00:47 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/constant-fold.mlir
} func.func @RemoveTrivialAdd(%arg0: tensor<2x2xf32>) -> tensor<2x2xf32> { %cst = arith.constant dense<0.0> : tensor<2x2xf32> %0 = "tf.Add"(%arg0, %cst) : (tensor<2x2xf32>, tensor<2x2xf32>) -> tensor<2x2xf32> func.return %0 : tensor<2x2xf32> // CHECK-LABEL: RemoveTrivialAdd // CHECK-NEXT: return %arg0 : tensor<2x2xf32> } func.func @RemoveTrivialAddBf16RHS(%arg0: tensor<2x2xbf16>) -> 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/lite/stablehlo/tests/stablehlo-custom-call-legalize-composite.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 19:49:44 UTC 2024 - 1009 bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/lift_quantizable_spots_as_functions_with_quantization_specs.mlir
// RUN: -split-input-file | FileCheck %s --check-prefix=STATIC-RANGE-PTQ-TO-COMPUTE-HEAVY // STATIC-RANGE-PTQ-TO-COMPUTE-HEAVY: @main func.func @main(%arg0: tensor<1x2xf32>) -> tensor<1x2xf32> { %0 = stablehlo.add %arg0, %arg0 : tensor<1x2xf32> return %0 : tensor<1x2xf32> } // Tests that `composite_add_fn_1` does not quantize when quantizing // only compute-heavy ops.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 02 18:09:38 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/verify_input_dialect_to_executor_pass_test.mlir
func.func @testNoClusterFuncOpPasses(%arg0: tensor<4x?x!tf_type.stringref>) -> tensor<4x2x!tf_type.string> { %0 = "tf.Identity"(%arg0) : (tensor<4x?x!tf_type.stringref>) -> tensor<4x2x!tf_type.string> func.return %0 : tensor<4x2x!tf_type.string> } // ----- func.func @_func(%arg0: tensor<i32>) -> tensor<i32> { func.return %arg0 : tensor<i32> } func.func @testClusterFuncOpFails(%arg0: tensor<i32>) -> tensor<i32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 30 22:07:53 UTC 2024 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/composite-lowering.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 18:45:51 UTC 2024 - 32.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/quantize_composite_functions_weight_only.mlir
} func.func private @composite_dot_general_fn(%arg0: tensor<1x2xf32>, %arg1: tensor<2x3xf32>) -> tensor<1x3xf32> attributes {_from_xla_call_module} { %0 = stablehlo.dot_general %arg0, %arg1, contracting_dims = [1] x [0] : (tensor<1x2xf32>, tensor<2x3xf32>) -> tensor<1x3xf32> return %0 : tensor<1x3xf32> } } // CHECK-LABEL: quantize_dot_general_per_tensor // CHECK-SAME: %[[ARG0:.+]]: tensor<1x2xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 05:56:10 UTC 2024 - 9.4K bytes - Viewed (0)