- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 56 for PartitionedCall (0.24 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/tests/add_dump_tensor_op.mlir
%0 = "tf.PartitionedCall"(%arg0, %cst, %cst_0) {config = "", config_proto = "", executor_type = "", f = @composite_conv2d_with_bias_and_relu6_fn_2} : (tensor<1x2x2x3xf32>, tensor<2x2x3x2xf32>, tensor<2xf32>) -> tensor<*xf32> loc(callsite("test@conv"("Conv2D") at "QuantizationUnit(\12\06Conv2D\1a\04conv)"))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 22:55:22 UTC 2024 - 37.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/lift_quantizable_spots_as_functions.mlir
} // CHECK-LABEL: @float_conv_3 // CHECK: "tf.PartitionedCall" // CHECK-SAME: f = @composite_conv2d_with_bias_and_relu6_fn_6 // CHECK: "tf.PartitionedCall" // CHECK-SAME: f = @composite_conv2d_with_bias_and_relu6_fn_5 // CHECK-LABEL: @float_conv_1 // CHECK: "tf.PartitionedCall" // CHECK-SAME: f = @composite_conv2d_with_bias_and_relu6_fn_2 // CHECK: "tf.PartitionedCall"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 26.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_composite_functions_xla.mlir
// CHECK: %[[quantized_gather:.*]] = "tf.PartitionedCall"( // CHECK-SAME: f = @quantized_gather_hybrid_fn_0 // CHECK: %[[float_conv:.*]] = "tf.PartitionedCall"(%[[quantized_gather]] // CHECK-SAME: f = @composite_conv2d_fn_2 // CHECK: %[[quantize:.*]] = "tf.PartitionedCall"(%[[float_conv]] // CHECK-SAME: f = @quantize_i8 // CHECK: %[[quantized_conv:.*]] = "tf.PartitionedCall"(%[[quantize]]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jan 08 01:16:10 UTC 2024 - 25.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_composite_functions_drq.mlir
%weight = "tf.Const"() {value = dense<2.000000e+00> : tensor<2x3x3x2xf32>} : () -> tensor<2x3x3x2xf32> %1 = "tf.PartitionedCall"(%arg0, %weight) {_tfl_quant_trait = "fully_quantizable", config = "", config_proto = "", executor_type = "", f = @composite_conv2d_fn_1} : (tensor<1x2x2x3xf32>, tensor<2x3x3x2xf32>) -> tensor<*xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 05 18:35:42 UTC 2024 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize-skip-partitioned-calls.mlir
%1 = "tf.PartitionedCall"(%0) <{ config = "", config_proto = "", executor_type = "", f = @some_other_func }> { _collective_manager_ids = [], device = "" } : (tensor<1x2x2x3xf32>) -> tensor<1x2x2x3xf32> // CHECK-SKIP: tf.PartitionedCall // CHECK-NOSKIP: call @some_other_func // CHECK-NOSKIP-NOT: tf.PartitionedCall func.return %1: tensor<1x2x2x3xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 08 20:05:12 UTC 2024 - 1.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantized_function_library.mlir
%i8_max = "tf.Const"() {value = dense<127> : tensor<i8>} : () -> tensor<i8> %clip_min = "tf.PartitionedCall"(%i8_min, %out_scale, %out_zp) { config = "", config_proto = "", executor_type = "", f=@internal_dequantize_i8_fn } : (tensor<i8>, tensor<*xf32>, tensor<*xi32>) -> tensor<*xf32> %clip_max = "tf.PartitionedCall"(%i8_max, %out_scale, %out_zp) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jan 08 01:16:10 UTC 2024 - 30.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/lift_hashtable_ops_as_args.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 15 05:41:44 UTC 2024 - 13.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/insert_custom_aggregation_ops.mlir
module { func.func @wrap_composite_func(%arg0: tensor<*xf32>, %arg1: tensor<*xf32>) -> tensor<*xf32> { %0 = "tf.PartitionedCall"(%arg0, %arg1) <{f = @composite_conv2d_with_relu6_fn}> {_tfl_quant_trait = "fully_quantizable"} : (tensor<*xf32>, tensor<*xf32>) -> tensor<*xf32> func.return %0 : tensor<*xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 32.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_drq.mlir
// ----- module { func.func @matmul(%arg0: tensor<1x2x2x3xf32>) -> (tensor<*xf32>) { %cst_0 = "tf.Const"() {value = dense<0.000000e+00> : tensor<2x1024xf32>} : () -> tensor<2x1024xf32> %1 = "tf.PartitionedCall"(%arg0, %cst_0) {_tfl_quant_trait = "fully_quantizable", config = "", config_proto = "", executor_type = "", f = @composite_matmul_fn} : (tensor<1x2x2x3xf32>, tensor<2x1024xf32>) -> tensor<*xf32> func.return %1: tensor<*xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:32:28 UTC 2024 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.mlir
// CHECK: return %[[SUBGRAPH_1]] : tensor<1024x3xf32> // CHECK: } } // ----- // main function contains PartitionedCall and StatefulPartitionedCall ops which // is used to preserve aliased functions. This test make sure stablehlo ops in // each PartitionedCall functions are lifted. module attributes {tf.versions = {bad_consumers = [], min_consumer = 12 : i32, producer = 1629 : i32}, tf_saved_model.semantics} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 01:09:50 UTC 2024 - 39.8K bytes - Viewed (0)