- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 18 for num_batch_threads (0.36 sec)
-
tensorflow/compiler/mlir/tfrt/tests/reconfig_batch_op.mlir
// CHECK-SAME: num_batch_threads = 2 : i64 // CHECK-SAME: shared_name = "batch/"
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/tfrt/tests/tfrt_fallback/batching_fallback.mlir
// Two batch_size=2 batches get concatenated. %result_1 = tfrt_fallback_async.batch_function device("/device:CPU:0") @matmul_cpu (%a1, %b) { num_batch_threads = 1, max_batch_size = 4, allowed_batch_sizes = [4], batch_timeout_micros = 1000000, container = "container", shared_name = "shared_name", batching_queue = "batching_queue",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jul 18 22:58:56 UTC 2023 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/resource_analyzer.mlir
%1 = "tf.BatchFunction"(%0) { f = @called_by_batch_func, operandSegmentSizes = array<i32: 1, 0>, batch_timeout_micros = 1000, max_batch_size = 8, num_batch_threads = 2 } : (tensor<!tf_type.resource<tensor<2xi32>>>) -> tensor<*xi32> return %1 : tensor<*xi32> } func.func private @called_by_batch_func(%arg: tensor<*x!tf_type.resource>) -> tensor<?xi32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/batch_function_deduplicate.mlir
func.return %0#0 : tensor<*xi32> } // CHECK: func private @compute_0
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/batch_function_deduplicate_failed.mlir
func.return %0#0 : tensor<*xi32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/force_shared_name_for_resource_ops.pbtxt
value: { func: { name: "inference_function" } } } attr: { key: "max_batch_size" value: { i: 32 } } attr: { key: "num_batch_threads" value: { i: 16 } } attr: { key: "shared_name" value: { s: "" } } } library { function { signature {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Aug 31 02:37:48 UTC 2021 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/sink_in_invariant_ops.mlir
func.return %1 : tensor<*xf32> } } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 21K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/batch_function_lowering.mlir
// CHECK-SAME: num_batch_threads = 1 : i64 // CHECK-SAME: shared_name = "batch/"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/batch_function_fallback_resource_variable_as_captured_tensor.mlir
%outputs_2, %control_2 = tf_executor.island wraps "tf.BatchFunction"(%outputs_1) {batch_timeout_micros = 5000 : i64, batching_queue = "", container = "", f = @batched_func, max_batch_size = 256 : i64, num_batch_threads = 2 : i64, operandSegmentSizes = array<i32: 0, 1>, shared_name = ""} : (tensor<*x!tf_type.resource>) -> tensor<*xf32> tf_executor.fetch %outputs_2 : tensor<*xf32> } func.return %0 : tensor<*xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/convert_tpu_model_to_cpu.mlir
// ----- // Tests that `tf.BatchFunction` is inlined. func.func @serving_default(%arg0: tensor<1xf32>, %arg1: tensor<1xf32>) -> tensor<1xf32> { %0 = "tf.BatchFunction"(%arg0, %arg1) {f = @batched_func, num_batch_threads = 1 : i64, max_batch_size = 2 : i64, batch_timeout_micros = 10000 : i64, operandSegmentSizes = array<i32: 1, 1>} : (tensor<1xf32>, tensor<1xf32>) -> (tensor<1xf32>) return %0 : tensor<1xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 4.3K bytes - Viewed (0)