- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 40 for PartitionedCall (0.16 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/tests/components/tf_to_stablehlo.mlir
// ----- // This test makes sure functions without tf._noinline=true is inlined. module { func.func @partitioned_call(%arg0: tensor<1x2x2x3xf32>) -> (tensor<1x2x2x3xf32>) { %0 = "tf.PartitionedCall"(%arg0) <{ config = "", config_proto = "", executor_type = "", f = @some_func }> { _collective_manager_ids = [], device = "" } : (tensor<1x2x2x3xf32>) -> tensor<1x2x2x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 08 20:05:12 UTC 2024 - 13.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/attribute_utils.cc
auto value = device_attr.getValue(); // TODO(b/229028654): Remove string conversion once we have C++17. absl::string_view device_type(value.data(), value.size()); // Device type may be empty for some ops, e.g. tf.PartitionedCall. auto it = std::find(kValidDeviceTypes.begin(), kValidDeviceTypes.end(), device_type); if (it == kValidDeviceTypes.end()) return failure(); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/decompose_optionals.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tpu_model_to_cpu.cc
return failure(); } rewriter.setInsertionPointAfter(call_op); // The TPUPartitionedCall has a TPUOrdinalSelectorOp for its last argument // which should be removed. So the replaced PartitionedCall op should keep // its original arguments except for the last element. SmallVector<Value> args = call_op.getOperands().drop_back(); rewriter.replaceOpWithNewOp<TF::PartitionedCallOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize.mlir
// CHECK-NEXT: [[q_bias:%.+]] = "quantfork.qcast"([[bias]]) : (tensor<2xf32>) -> tensor<2x!quant.uniform<i32:f32, 0.044022349891595126>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:32:28 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc
return true; } } return false; } // Checks if the function is only used by supported ops. Returns false when the // function has no uses. Currently, only PartitionedCall is supported. // TODO(b/284222309): Support lifting for functions called by control flow. bool UsedBySupportedOps(ModuleOp module, func::FuncOp func) { auto function_uses =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/prepare_tpu_computation_for_tf_export.cc
// Collect all the ops that needs to have token input names attributes. These // ops are communication ops and all their parent ops via nesting or function // calls. For example, IfRegion op and PartitionedCall op. std::vector<Operation*> worklist; absl::flat_hash_set<Operation*> ops_with_tokens; module.walk([&](Operation* op) { if (IsCommunicationOp(op)) { ops_with_tokens.insert(op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/clustering_bridge_passes.cc
// Guarantee all functions have one use, which enables more exact shape // inference. pm.addPass(mlir::TF::CreateGuaranteeAllFuncsOneUsePass()); pm.addPass(mlir::TF::CreateTFShapeInferencePass()); // Encapsulate PartitionedCall ops within a cluster so that the composite // resource ops can be decomposed. pm.addPass(tensorflow::tf2xla::internal::CreateXlaClusterFormationPass());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 16:09:14 UTC 2024 - 11.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
for (int i = 0; i < num_results; ++i) { main_func.setResultAttr( i, kTfSavedModelIndexPathAttr, ArrayAttr::get(context, {StringAttr::get(context, output_names[i])})); } // Creates PartitionedCall ops to call exported functions. auto guard = OpBuilder::InsertionGuard(builder); int arg_idx = 0; int result_idx = 0; llvm::SmallVector<Value> call_op_returns;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.td
} def XlaClusterFormationPass : Pass<"tf-xla-cluster-formation", "ModuleOp"> { let summary = "Encapsulate partitioned calls within a Cluster op"; let description = [{ This pass clusters `tf.PartitionedCall` and `tf.StatefulPartitionedCall` with `_xla_compile_device_type` attribute into a `tf_device.cluster`. Notice this pass will only rewrite the outermost call if there are nested
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 19.8K bytes - Viewed (0)