- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for BatchFunction (1.07 sec)
-
tensorflow/compiler/mlir/tfrt/tests/reconfig_batch_op.mlir
%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] // CHECK-SAME: batch_timeout_micros = 100000 : i64 // CHECK-SAME: batching_queue = "" // CHECK-SAME: container = "" // CHECK-SAME: enable_large_batch_splitting = false
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/batch_function_lowering.mlir
func.return %2 : tensor<1x3xf32> } // CHECK-LABEL: func @main func.func @main(%arg0: tensor<1x3xf32>) -> tensor<*xf32> attributes {tf.entry_function = {control_outputs = "", inputs = "input:0", outputs = "batch/BatchFunction:0"}} { %0 = "tf.VarHandleOp"() {device = "/device:CPU:0", container = "", shared_name = "variable"} : () -> tensor<!tf_type.resource<tensor<1x3xf32>>>
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/tensorflow/translate/upgrade_graph.cc
namespace { // Returns the ops that should use node name if shared_name is empty. const llvm::StringSet<>& GetOpsUsingNodeName() { static auto* const ops = new llvm::StringSet<>({"VariableV2", "Variable", "BatchFunction"}); return *ops; } // Returns the set of ops that we want to generate shared_names for them if // empty. const llvm::StringSet<>& GetSharedNameGenerationCompatibleOps() { return GetOpsUsingNodeName(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 14:33:47 UTC 2024 - 6.2K bytes - Viewed (0)